initial implementation for fluid interface check

dosisod · Dec 9, 2023 · b823ce9 · b823ce9
1 parent 70c85e8
commit b823ce9
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/test/data/err_999.py b/test/data/err_999.py
@@ -30,6 +30,10 @@ class functions:
         def lit(constant):
             return constant
 
+        @staticmethod
+        def col(col_name):
+            return col_name
+
     class DataFrame:
         @staticmethod
         def withColumnRenamed(col_in, col_out):
@@ -63,6 +67,14 @@ def process(file_name: str):
     return df
 
 
+def projection(df_in: spark.DataFrame) -> spark.DataFrame:
+    df = (
+        df_in.select(["col1", "col2"])
+        .withColumnRenamed("col1", "col1a")
+    )
+    return df.withColumn("col2a", spark.functions.col("col2").cast("date"))
+
+
 # these will not
 def no_match():
     y = 10