Fix tests

tests/integ/scala/test_udtf_suite.py

@@ -218,7 +218,7 @@ def process(self) -> int:
     with pytest.raises(ValueError) as ve:
         session.udtf.register(MyWrongReturnTypeUDTF, output_schema=["c1"])
     assert (
-        "The return type hint for a UDTF handler must but a collection type or a PandasDataFrame. <class 'int'> is used."
+        "The return type hint for a UDTF handler must be a collection type or a PandasDataFrame. <class 'int'> is used."
         in str(ve.value)
     )
 

tests/integ/test_udtf.py

@@ -40,9 +40,6 @@
 
 @pytest.fixture(scope="module")
 def vectorized_udtf_test_table(session) -> str:
-    session.sql(
-        "alter session set python_udtf_enable_end_partition_dataframe_encoding = true;"
-    ).collect()
     # Input tabular data
     table_name = Utils.random_table_name()
     session.create_dataframe(

tests/unit/test_types.py

@@ -605,13 +605,16 @@ def {func_name}(x: collections.defaultdict, y: Union[datetime.date, time]) -> Op
 def {func_name}_{func_name}(x: int) -> int:
     return x
 """
-    with pytest.raises(ValueError, match="function.*is not found in file"):
-        retrieve_func_type_hints_from_source("", func_name, _source=source)
+    # Func not found in file
+    assert retrieve_func_type_hints_from_source("", func_name, _source=source) is None
 
-    with pytest.raises(ValueError, match="class.*is not found in file"):
+    # Class not found in file
+    assert (
         retrieve_func_type_hints_from_source(
             "", func_name, class_name="FakeClass", _source=source
         )
+        is None
+    )
 
     source = f"""
 def {func_name}() -> 1: