snowpark-connect 0.24.0__py3-none-any.whl → 0.26.0__py3-none-any.whl

snowflake/snowpark_connect/includes/python/pyspark/errors/tests/__init__.py

@@ -1,16 +0,0 @@
-#
-# Licensed to the Apache Software Foundation (ASF) under one or more
-# contributor license agreements.  See the NOTICE file distributed with
-# this work for additional information regarding copyright ownership.
-# The ASF licenses this file to You under the Apache License, Version 2.0
-# (the "License"); you may not use this file except in compliance with
-# the License.  You may obtain a copy of the License at
-#
-#    http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-#

snowflake/snowpark_connect/includes/python/pyspark/errors/tests/test_errors.py

@@ -1,60 +0,0 @@
-# -*- encoding: utf-8 -*-
-#
-# Licensed to the Apache Software Foundation (ASF) under one or more
-# contributor license agreements.  See the NOTICE file distributed with
-# this work for additional information regarding copyright ownership.
-# The ASF licenses this file to You under the Apache License, Version 2.0
-# (the "License"); you may not use this file except in compliance with
-# the License.  You may obtain a copy of the License at
-#
-#    http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-#
-
-import json
-import unittest
-
-from pyspark.errors.error_classes import ERROR_CLASSES_JSON
-from pyspark.errors.utils import ErrorClassesReader
-
-
-class ErrorsTest(unittest.TestCase):
-    def test_error_classes_sorted(self):
-        # Test error classes is sorted alphabetically
-        error_reader = ErrorClassesReader()
-        error_class_names = list(error_reader.error_info_map.keys())
-        for i in range(len(error_class_names) - 1):
-            self.assertTrue(
-                error_class_names[i] < error_class_names[i + 1],
-                f"Error class [{error_class_names[i]}] should place"
-                f"after [{error_class_names[i + 1]}]",
-            )
-
-    def test_error_classes_duplicated(self):
-        # Test error classes is not duplicated
-        def detect_duplication(pairs):
-            error_classes_json = {}
-            for name, message in pairs:
-                self.assertTrue(name not in error_classes_json, f"Duplicate error class: {name}")
-                error_classes_json[name] = message
-            return error_classes_json
-
-        json.loads(ERROR_CLASSES_JSON, object_pairs_hook=detect_duplication)
-
-
-if __name__ == "__main__":
-    import unittest
-    from pyspark.errors.tests.test_errors import *  # noqa: F401
-
-    try:
-        import xmlrunner
-
-        testRunner = xmlrunner.XMLTestRunner(output="target/test-reports", verbosity=2)
-    except ImportError:
-        testRunner = None
-    unittest.main(testRunner=testRunner, verbosity=2)

snowflake/snowpark_connect/includes/python/pyspark/ml/deepspeed/tests/test_deepspeed_distributor.py

@@ -1,306 +0,0 @@
-# mypy: ignore-errors
-# Licensed to the Apache Software Foundation (ASF) under one or more
-# contributor license agreements.  See the NOTICE file distributed with
-# this work for additional information regarding copyright ownership.
-# The ASF licenses this file to You under the Apache License, Version 2.0
-# (the "License"); you may not use this file except in compliance with
-# the License.  You may obtain a copy of the License at
-#
-#    http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-#
-from contextlib import contextmanager
-import os
-import shutil
-import sys
-import textwrap
-from typing import Any, Callable, Dict, Tuple
-import unittest
-
-from pyspark import SparkConf, SparkContext
-from pyspark.ml.deepspeed.deepspeed_distributor import DeepspeedTorchDistributor
-from pyspark.sql import SparkSession
-from pyspark.ml.torch.tests.test_distributor import (
-    get_local_mode_conf,
-    set_up_test_dirs,
-    get_distributed_mode_conf,
-)
-
-have_deepspeed = True
-try:
-    import deepspeed  # noqa: F401
-except ImportError:
-    have_deepspeed = False
-
-
-class DeepspeedTorchDistributorUnitTests(unittest.TestCase):
-    def _get_env_var(self, var_name: str, default_value: Any) -> Any:
-        value = os.getenv(var_name)
-        if value:
-            return value
-        os.environ[var_name] = str(default_value)
-        return default_value
-
-    def _get_env_variables_distributed(self) -> Tuple[Any, Any, Any]:
-        master_addr = self._get_env_var("MASTER_ADDR", "127.0.0.1")
-        master_port = self._get_env_var("MASTER_PORT", 2000)
-        rank = self._get_env_var("RANK", 0)
-        return master_addr, master_port, rank
-
-    def test_get_torchrun_args_local(self) -> None:
-        number_of_processes = 5
-        expected_torchrun_args_local = ["--standalone", "--nnodes=1"]
-        expected_processes_per_node_local = number_of_processes
-        (
-            get_local_mode_torchrun_args,
-            process_per_node,
-        ) = DeepspeedTorchDistributor._get_torchrun_args(True, number_of_processes)
-        self.assertEqual(get_local_mode_torchrun_args, expected_torchrun_args_local)
-        self.assertEqual(expected_processes_per_node_local, process_per_node)
-
-    def test_get_torchrun_args_distributed(self) -> None:
-        number_of_processes = 5
-        master_addr, master_port, rank = self._get_env_variables_distributed()
-        expected_torchrun_args_distributed = [
-            f"--nnodes={number_of_processes}",
-            f"--node_rank={rank}",
-            f"--rdzv_endpoint={master_addr}:{master_port}",
-            "--rdzv_id=0",
-        ]
-        torchrun_args_distributed, process_per_node = DeepspeedTorchDistributor._get_torchrun_args(
-            False, number_of_processes
-        )
-        self.assertEqual(torchrun_args_distributed, expected_torchrun_args_distributed)
-        self.assertEqual(process_per_node, 1)
-
-    def test_create_torchrun_command_local(self) -> None:
-        deepspeed_conf = "path/to/deepspeed"
-        train_file_path = "path/to/exec"
-        num_procs = 10
-        input_params: Dict[str, Any] = {}
-        input_params["local_mode"] = True
-        input_params["num_processes"] = num_procs
-        input_params["deepspeed_config"] = deepspeed_conf
-
-        torchrun_local_args_expected = ["--standalone", "--nnodes=1"]
-        with self.subTest(msg="Testing local training with no extra args"):
-            local_cmd_no_args_expected = [
-                sys.executable,
-                "-m",
-                "torch.distributed.run",
-                *torchrun_local_args_expected,
-                f"--nproc_per_node={num_procs}",
-                train_file_path,
-                "--deepspeed",
-                "--deepspeed_config",
-                deepspeed_conf,
-            ]
-            local_cmd = DeepspeedTorchDistributor._create_torchrun_command(
-                input_params, train_file_path
-            )
-            self.assertEqual(local_cmd, local_cmd_no_args_expected)
-        with self.subTest(msg="Testing local training with extra args for the training script"):
-            local_mode_version_args = ["--arg1", "--arg2"]
-            local_cmd_args_expected = [
-                sys.executable,
-                "-m",
-                "torch.distributed.run",
-                *torchrun_local_args_expected,
-                f"--nproc_per_node={num_procs}",
-                train_file_path,
-                *local_mode_version_args,
-                "--deepspeed",
-                "--deepspeed_config",
-                deepspeed_conf,
-            ]
-
-            local_cmd_with_args = DeepspeedTorchDistributor._create_torchrun_command(
-                input_params, train_file_path, *local_mode_version_args
-            )
-            self.assertEqual(local_cmd_with_args, local_cmd_args_expected)
-
-    def test_create_torchrun_command_distributed(self) -> None:
-        deepspeed_conf = "path/to/deepspeed"
-        train_file_path = "path/to/exec"
-        num_procs = 10
-        input_params: Dict[str, Any] = {}
-        input_params["local_mode"] = True
-        input_params["num_processes"] = num_procs
-        input_params["deepspeed_config"] = deepspeed_conf
-        (
-            distributed_master_address,
-            distributed_master_port,
-            distributed_rank,
-        ) = self._get_env_variables_distributed()
-        distributed_torchrun_args = [
-            f"--nnodes={num_procs}",
-            f"--node_rank={distributed_rank}",
-            f"--rdzv_endpoint={distributed_master_address}:{distributed_master_port}",
-            "--rdzv_id=0",
-        ]
-        with self.subTest(msg="Distributed training command verification with no extra args"):
-            distributed_cmd_no_args_expected = [
-                sys.executable,
-                "-m",
-                "torch.distributed.run",
-                *distributed_torchrun_args,
-                "--nproc_per_node=1",
-                train_file_path,
-                "--deepspeed",
-                "--deepspeed_config",
-                deepspeed_conf,
-            ]
-            input_params["local_mode"] = False
-            distributed_command = DeepspeedTorchDistributor._create_torchrun_command(
-                input_params, train_file_path
-            )
-            self.assertEqual(distributed_cmd_no_args_expected, distributed_command)
-        with self.subTest(msg="Distributed training command verification with extra arguments"):
-            distributed_extra_args = ["-args1", "--args2"]
-            distributed_cmd_args_expected = [
-                sys.executable,
-                "-m",
-                "torch.distributed.run",
-                *distributed_torchrun_args,
-                "--nproc_per_node=1",
-                train_file_path,
-                *distributed_extra_args,
-                "--deepspeed",
-                "--deepspeed_config",
-                deepspeed_conf,
-            ]
-            distributed_command_with_args = DeepspeedTorchDistributor._create_torchrun_command(
-                input_params, train_file_path, *distributed_extra_args
-            )
-            self.assertEqual(distributed_cmd_args_expected, distributed_command_with_args)
-
-
-def _create_basic_function() -> Callable:
-    # TODO: swap out with better test function
-    # once Deepspeed better supports CPU
-    def pythagoras(leg1: float, leg2: float) -> float:
-        import deepspeed
-
-        print(deepspeed.__version__)
-        return (leg1 * leg1 + leg2 * leg2) ** 0.5
-
-    return pythagoras
-
-
-@contextmanager
-def _create_pytorch_training_test_file():
-    # Note: when Deepspeed CPU support becomes better,
-    # switch in more realistic training files using Deepspeed
-    # optimizations + constructs
-    str_to_write = textwrap.dedent(
-        """ 
-            import sys
-            def pythagorean_thm(x : int, y: int): # type: ignore 
-                import deepspeed # type: ignore
-                return (x*x + y*y)**0.5 # type: ignore
-            print(pythagorean_thm(int(sys.argv[1]), int(sys.argv[2])))"""
-    )
-    cp_path = "/tmp/test_deepspeed_training_file.py"
-    with open(cp_path, "w") as f:
-        f.write(str_to_write)
-    yield cp_path
-    os.remove(cp_path)
-
-
-# The program and function that we use in the end-to-end tests
-# is very simple because in the Spark CI we only have access
-# to CPUs and at this point in time, CPU support is limited
-# in Deepspeed. Once Deepspeed better supports CPU training
-# and inference, the hope is to switch out the training
-# and file for the tests with more realistic testing
-# that use Deepspeed constructs.
-@unittest.skipIf(not have_deepspeed, "deepspeed is required for these tests")
-class DeepspeedTorchDistributorDistributedEndToEnd(unittest.TestCase):
-    @classmethod
-    def setUpClass(cls) -> None:
-        (cls.gpu_discovery_script_file_name, cls.mnist_dir_path) = set_up_test_dirs()  # noqa
-        # "loadDefaults" is set to False because if not, the SparkConf will
-        # use contain configurations from the LocalEndToEnd test,
-        # which causes the test to break.
-        conf = SparkConf(loadDefaults=False)
-        for k, v in get_distributed_mode_conf().items():
-            conf = conf.set(k, v)
-        conf = conf.set(
-            "spark.worker.resource.gpu.discoveryScript", cls.gpu_discovery_script_file_name
-        )
-        sc = SparkContext("local-cluster[2,2,512]", cls.__name__, conf=conf)
-        cls.spark = SparkSession(sc)
-
-    @classmethod
-    def tearDownClass(cls) -> None:
-        shutil.rmtree(cls.mnist_dir_path)
-        os.unlink(cls.gpu_discovery_script_file_name)
-        cls.spark.stop()
-
-    def test_simple_function_e2e(self) -> None:
-        train_fn = _create_basic_function()
-        # Arguments for the pythagoras function train_fn
-        x = 3
-        y = 4
-        dist = DeepspeedTorchDistributor(numGpus=2, useGpu=False, localMode=False)
-        output = dist.run(train_fn, x, y)
-        self.assertEqual(output, 5)
-
-    def test_pytorch_file_e2e(self) -> None:
-        # TODO: change to better test script
-        # once Deepspeed CPU support is better
-        with _create_pytorch_training_test_file() as cp_path:
-            dist = DeepspeedTorchDistributor(numGpus=True, useGpu=False, localMode=False)
-            dist.run(cp_path, 2, 5)
-
-
-@unittest.skipIf(not have_deepspeed, "deepspeed is required for these tests")
-class DeepspeedDistributorLocalEndToEndTests(unittest.TestCase):
-    @classmethod
-    def setUpClass(cls) -> None:
-        cls.gpu_discovery_script_file_name, cls.mnist_dir_path = set_up_test_dirs()  # noqa
-        conf = SparkConf()
-        for k, v in get_local_mode_conf().items():
-            conf = conf.set(k, v)
-        conf = conf.set(
-            "spark.driver.resource.gpu.discoveryScript", cls.gpu_discovery_script_file_name
-        )
-        sc = SparkContext("local-cluster[2,2,512]", cls.__name__, conf=conf)
-        cls.spark = SparkSession(sc)
-
-    @classmethod
-    def tearDownClass(cls) -> None:
-        shutil.rmtree(cls.mnist_dir_path)
-        os.unlink(cls.gpu_discovery_script_file_name)
-        cls.spark.stop()
-
-    def test_simple_function_e2e(self) -> None:
-        train_fn = _create_basic_function()
-        # Arguments for the pythagoras function train_fn
-        x = 3
-        y = 4
-        dist = DeepspeedTorchDistributor(numGpus=2, useGpu=False, localMode=True)
-        output = dist.run(train_fn, x, y)
-        self.assertEqual(output, 5)
-
-    def test_pytorch_file_e2e(self) -> None:
-        with _create_pytorch_training_test_file() as path_to_train_file:
-            dist = DeepspeedTorchDistributor(numGpus=2, useGpu=False, localMode=True)
-            dist.run(path_to_train_file, 2, 5)
-
-
-if __name__ == "__main__":
-    from pyspark.ml.deepspeed.tests.test_deepspeed_distributor import *  # noqa: F401,F403
-
-    try:
-        import xmlrunner  # type:ignore
-
-        testRunner = xmlrunner.XMLTestRunner(output="target/test-reports", verbosity=2)
-    except ImportError:
-        testRunner = None
-    unittest.main(testRunner=testRunner, verbosity=2)

snowflake/snowpark_connect/includes/python/pyspark/ml/tests/__init__.py

@@ -1,16 +0,0 @@
-#
-# Licensed to the Apache Software Foundation (ASF) under one or more
-# contributor license agreements.  See the NOTICE file distributed with
-# this work for additional information regarding copyright ownership.
-# The ASF licenses this file to You under the Apache License, Version 2.0
-# (the "License"); you may not use this file except in compliance with
-# the License.  You may obtain a copy of the License at
-#
-#    http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-#

snowflake/snowpark_connect/includes/python/pyspark/ml/tests/connect/test_connect_classification.py

@@ -1,53 +0,0 @@
-# -*- coding: utf-8 -*-
-#
-# Licensed to the Apache Software Foundation (ASF) under one or more
-# contributor license agreements.  See the NOTICE file distributed with
-# this work for additional information regarding copyright ownership.
-# The ASF licenses this file to You under the Apache License, Version 2.0
-# (the "License"); you may not use this file except in compliance with
-# the License.  You may obtain a copy of the License at
-#
-#    http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-#
-import unittest
-import os
-
-from pyspark.sql import SparkSession
-from pyspark.ml.tests.connect.test_legacy_mode_classification import ClassificationTestsMixin
-
-have_torch = "SPARK_SKIP_CONNECT_COMPAT_TESTS" not in os.environ
-try:
-    import torch  # noqa: F401
-except ImportError:
-    have_torch = False
-
-
-@unittest.skipIf(not have_torch, "torch is required")
-class ClassificationTestsOnConnect(ClassificationTestsMixin, unittest.TestCase):
-    def setUp(self) -> None:
-        self.spark = (
-            SparkSession.builder.remote(os.environ.get("SPARK_CONNECT_TESTING_REMOTE", "local[2]"))
-            .config("spark.connect.copyFromLocalToFs.allowDestLocal", "true")
-            .getOrCreate()
-        )
-
-    def tearDown(self) -> None:
-        self.spark.stop()
-
-
-if __name__ == "__main__":
-    from pyspark.ml.tests.connect.test_connect_classification import *  # noqa: F401,F403
-
-    try:
-        import xmlrunner  # type: ignore[import]
-
-        testRunner = xmlrunner.XMLTestRunner(output="target/test-reports", verbosity=2)
-    except ImportError:
-        testRunner = None
-    unittest.main(testRunner=testRunner, verbosity=2)

snowflake/snowpark_connect/includes/python/pyspark/ml/tests/connect/test_connect_evaluation.py

@@ -1,50 +0,0 @@
-#
-# Licensed to the Apache Software Foundation (ASF) under one or more
-# contributor license agreements.  See the NOTICE file distributed with
-# this work for additional information regarding copyright ownership.
-# The ASF licenses this file to You under the Apache License, Version 2.0
-# (the "License"); you may not use this file except in compliance with
-# the License.  You may obtain a copy of the License at
-#
-#    http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-#
-import os
-import unittest
-
-from pyspark.sql import SparkSession
-from pyspark.ml.tests.connect.test_legacy_mode_evaluation import EvaluationTestsMixin
-
-have_torcheval = "SPARK_SKIP_CONNECT_COMPAT_TESTS" not in os.environ
-try:
-    import torcheval  # noqa: F401
-except ImportError:
-    have_torcheval = False
-
-
-@unittest.skipIf(not have_torcheval, "torcheval is required")
-class EvaluationTestsOnConnect(EvaluationTestsMixin, unittest.TestCase):
-    def setUp(self) -> None:
-        self.spark = SparkSession.builder.remote(
-            os.environ.get("SPARK_CONNECT_TESTING_REMOTE", "local[2]")
-        ).getOrCreate()
-
-    def tearDown(self) -> None:
-        self.spark.stop()
-
-
-if __name__ == "__main__":
-    from pyspark.ml.tests.connect.test_connect_evaluation import *  # noqa: F401,F403
-
-    try:
-        import xmlrunner  # type: ignore[import]
-
-        testRunner = xmlrunner.XMLTestRunner(output="target/test-reports", verbosity=2)
-    except ImportError:
-        testRunner = None
-    unittest.main(testRunner=testRunner, verbosity=2)

snowflake/snowpark_connect/includes/python/pyspark/ml/tests/connect/test_connect_feature.py

@@ -1,43 +0,0 @@
-#
-# Licensed to the Apache Software Foundation (ASF) under one or more
-# contributor license agreements.  See the NOTICE file distributed with
-# this work for additional information regarding copyright ownership.
-# The ASF licenses this file to You under the Apache License, Version 2.0
-# (the "License"); you may not use this file except in compliance with
-# the License.  You may obtain a copy of the License at
-#
-#    http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-#
-import os
-import unittest
-
-from pyspark.sql import SparkSession
-from pyspark.ml.tests.connect.test_legacy_mode_feature import FeatureTestsMixin
-
-
-class FeatureTestsOnConnect(FeatureTestsMixin, unittest.TestCase):
-    def setUp(self) -> None:
-        self.spark = SparkSession.builder.remote(
-            os.environ.get("SPARK_CONNECT_TESTING_REMOTE", "local[2]")
-        ).getOrCreate()
-
-    def tearDown(self) -> None:
-        self.spark.stop()
-
-
-if __name__ == "__main__":
-    from pyspark.ml.tests.connect.test_connect_feature import *  # noqa: F401,F403
-
-    try:
-        import xmlrunner  # type: ignore[import]
-
-        testRunner = xmlrunner.XMLTestRunner(output="target/test-reports", verbosity=2)
-    except ImportError:
-        testRunner = None
-    unittest.main(testRunner=testRunner, verbosity=2)

snowflake/snowpark_connect/includes/python/pyspark/ml/tests/connect/test_connect_function.py

@@ -1,114 +0,0 @@
-#
-# Licensed to the Apache Software Foundation (ASF) under one or more
-# contributor license agreements.  See the NOTICE file distributed with
-# this work for additional information regarding copyright ownership.
-# The ASF licenses this file to You under the Apache License, Version 2.0
-# (the "License"); you may not use this file except in compliance with
-# the License.  You may obtain a copy of the License at
-#
-#    http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-#
-import os
-import unittest
-
-from pyspark.sql import SparkSession as PySparkSession
-from pyspark.sql.dataframe import DataFrame as SDF
-from pyspark.ml import functions as SF
-
-from pyspark.testing.sqlutils import SQLTestUtils
-from pyspark.testing.connectutils import (
-    should_test_connect,
-    ReusedConnectTestCase,
-)
-from pyspark.testing.pandasutils import PandasOnSparkTestUtils
-
-if should_test_connect:
-    from pyspark.sql.connect.dataframe import DataFrame as CDF
-    from pyspark.ml.connect import functions as CF
-
-
-@unittest.skipIf("SPARK_SKIP_CONNECT_COMPAT_TESTS" in os.environ, "Requires JVM access")
-class SparkConnectMLFunctionTests(ReusedConnectTestCase, PandasOnSparkTestUtils, SQLTestUtils):
-    """These test cases exercise the interface to the proto plan
-    generation but do not call Spark."""
-
-    @classmethod
-    def setUpClass(cls):
-        super(SparkConnectMLFunctionTests, cls).setUpClass()
-        # Disable the shared namespace so pyspark.sql.functions, etc point the regular
-        # PySpark libraries.
-        os.environ["PYSPARK_NO_NAMESPACE_SHARE"] = "1"
-        cls.connect = cls.spark  # Switch Spark Connect session and regular PySpark sesion.
-        cls.spark = PySparkSession._instantiatedSession
-        assert cls.spark is not None
-
-    @classmethod
-    def tearDownClass(cls):
-        cls.spark = cls.connect  # Stopping Spark Connect closes the session in JVM at the server.
-        super(SparkConnectMLFunctionTests, cls).setUpClass()
-        del os.environ["PYSPARK_NO_NAMESPACE_SHARE"]
-
-    def compare_by_show(self, df1, df2, n: int = 20, truncate: int = 20):
-        assert isinstance(df1, (SDF, CDF))
-        if isinstance(df1, SDF):
-            str1 = df1._jdf.showString(n, truncate, False)
-        else:
-            str1 = df1._show_string(n, truncate, False)
-
-        assert isinstance(df2, (SDF, CDF))
-        if isinstance(df2, SDF):
-            str2 = df2._jdf.showString(n, truncate, False)
-        else:
-            str2 = df2._show_string(n, truncate, False)
-
-        self.assertEqual(str1, str2)
-
-    def test_array_vector_conversion(self):
-        query = """
-            SELECT * FROM VALUES
-            (1, 4, ARRAY(1.0, 2.0, 3.0)),
-            (1, 2, ARRAY(-1.0, -2.0, -3.0))
-            AS tab(a, b, c)
-            """
-
-        cdf = self.connect.sql(query)
-        sdf = self.spark.sql(query)
-
-        self.compare_by_show(
-            cdf.select(cdf.b, CF.array_to_vector(cdf.c)),
-            sdf.select(sdf.b, SF.array_to_vector(sdf.c)),
-        )
-
-        cdf1 = cdf.select("a", CF.array_to_vector(cdf.c).alias("d"))
-        sdf1 = sdf.select("a", SF.array_to_vector(sdf.c).alias("d"))
-
-        self.compare_by_show(
-            cdf1.select(CF.vector_to_array(cdf1.d)),
-            sdf1.select(SF.vector_to_array(sdf1.d)),
-        )
-        self.compare_by_show(
-            cdf1.select(CF.vector_to_array(cdf1.d, "float32")),
-            sdf1.select(SF.vector_to_array(sdf1.d, "float32")),
-        )
-        self.compare_by_show(
-            cdf1.select(CF.vector_to_array(cdf1.d, "float64")),
-            sdf1.select(SF.vector_to_array(sdf1.d, "float64")),
-        )
-
-
-if __name__ == "__main__":
-    from pyspark.ml.tests.connect.test_connect_function import *  # noqa: F401
-
-    try:
-        import xmlrunner  # type: ignore
-
-        testRunner = xmlrunner.XMLTestRunner(output="target/test-reports", verbosity=2)
-    except ImportError:
-        testRunner = None
-    unittest.main(testRunner=testRunner, verbosity=2)