fix

Author: allisonwang-db

python/pyspark/sql/tests/arrow/test_arrow_udtf.py

@@ -931,30 +931,28 @@ def eval(
                 yield result_table
 
         # Test with DataFrame API using named arguments
-        # TODO(SPARK-53426): Support named table argument with DataFrame API
-        # input_df = self.spark.range(3)  # [0, 1, 2]
-        # result_df = NamedArgsUDTF(table_data=input_df.asTable(), multiplier=lit(5))
+        input_df = self.spark.range(3)  # [0, 1, 2]
+        result_df = NamedArgsUDTF(table_data=input_df.asTable(), multiplier=lit(5))
         expected_df = self.spark.createDataFrame(
             [(0, 5), (5, 5), (10, 5)], 
             "result_id bigint, multiplier_used int"
         )
-        # assertDataFrameEqual(result_df, expected_df)
+        assertDataFrameEqual(result_df, expected_df)
 
         # Test with DataFrame API using different named argument order
-        # TODO(SPARK-53426): Support named table argument with DataFrame API
-        # result_df2 = NamedArgsUDTF(multiplier=lit(3), table_data=input_df.asTable())
+        result_df2 = NamedArgsUDTF(multiplier=lit(3), table_data=input_df.asTable())
         expected_df2 = self.spark.createDataFrame(
             [(0, 3), (3, 3), (6, 3)], 
             "result_id bigint, multiplier_used int"
         )
-        # assertDataFrameEqual(result_df2, expected_df2)
+        assertDataFrameEqual(result_df2, expected_df2)
 
         # Test SQL registration and usage with named arguments
         self.spark.udtf.register("test_named_args_udtf", NamedArgsUDTF)
 
         sql_result_df = self.spark.sql("""
             SELECT * FROM test_named_args_udtf(
-                table_data => TABLE(SELECT id FROM range(0, 3))
+                table_data => TABLE(SELECT id FROM range(0, 3)),
                 multiplier => 5
             )
         """)