test

Author: zhengruifeng

python/pyspark/sql/connect/dataframe.py

@@ -1703,10 +1703,10 @@ def __getattr__(self, name: str) -> "Column":
                 errorClass="JVM_ATTRIBUTE_NOT_SUPPORTED", messageParameters={"attr_name": name}
             )
 
-        # if name not in self.columns:
-        #     raise PySparkAttributeError(
-        #         errorClass="ATTRIBUTE_NOT_SUPPORTED", messageParameters={"attr_name": name}
-        #     )
+        if name not in self.columns:
+            raise PySparkAttributeError(
+                errorClass="ATTRIBUTE_NOT_SUPPORTED", messageParameters={"attr_name": name}
+            )
 
         return self._col(name)
 
@@ -1739,14 +1739,14 @@ def __getitem__(
                 #   ConnectColumn(addDataFrameIdToCol(resolve(colName)))
                 # }
 
-                # # validate the column name
-                # if not hasattr(self._session, "is_mock_session"):
-                #     from pyspark.sql.connect.types import verify_col_name
-                #
-                #     # Try best to verify the column name with cached schema
-                #     # If fails, fall back to the server side validation
-                #     if not verify_col_name(item, self._schema):
-                #         self.select(item).isLocal()
+                # validate the column name
+                if not hasattr(self._session, "is_mock_session"):
+                    from pyspark.sql.connect.types import verify_col_name
+
+                    # Try best to verify the column name with cached schema
+                    # If fails, fall back to the server side validation
+                    if not verify_col_name(item, self._schema):
+                        self.select(item).isLocal()
 
                 return self._col(item)
         elif isinstance(item, Column):

sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/analysis/ColumnResolutionHelper.scala

@@ -34,6 +34,7 @@ import org.apache.spark.sql.connector.catalog.{CatalogManager, Identifier}
 import org.apache.spark.sql.errors.{DataTypeErrorsBase, QueryCompilationErrors}
 import org.apache.spark.sql.internal.SQLConf
 
+// scalastyle:off println
 trait ColumnResolutionHelper extends Logging with DataTypeErrorsBase {
 
   def conf: SQLConf