snowpark-connect 0.23.0__py3-none-any.whl → 0.25.0__py3-none-any.whl

snowflake/snowpark_connect/includes/python/pyspark/pandas/tests/data_type_ops/test_string_ops.py

@@ -1,354 +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 unittest
-
-import numpy as np
-import pandas as pd
-from pandas.api.types import CategoricalDtype
-
-from pyspark import pandas as ps
-from pyspark.pandas.config import option_context
-from pyspark.pandas.tests.data_type_ops.testing_utils import OpsTestBase
-from pyspark.pandas.typedef.typehints import extension_object_dtypes_available
-
-if extension_object_dtypes_available:
-    from pandas import StringDtype
-
-
-class StringOpsTestsMixin:
-    @property
-    def bool_pdf(self):
-        return pd.DataFrame({"this": ["x", "y", "z"], "that": ["z", "y", "x"]})
-
-    @property
-    def bool_psdf(self):
-        return ps.from_pandas(self.bool_pdf)
-
-    @property
-    def bool_non_numeric_pdf(self):
-        return pd.concat([self.bool_pdf, self.non_numeric_pdf], axis=1)
-
-    @property
-    def bool_non_numeric_psdf(self):
-        return ps.from_pandas(self.bool_non_numeric_pdf)
-
-    def test_add(self):
-        pdf, psdf = self.bool_non_numeric_pdf, self.bool_non_numeric_psdf
-        pser, psser = pdf["this"], psdf["this"]
-        other_pser, other_psser = pdf["that"], psdf["that"]
-        self.assert_eq(pser + "x", psser + "x")
-        self.assertRaises(TypeError, lambda: psser + 1)
-
-        self.assert_eq(pser + other_pser, psser + other_psser)
-
-        for col in self.non_numeric_df_cols:
-            if col != "string":
-                self.assertRaises(TypeError, lambda: psser + psdf[col])
-
-    def test_sub(self):
-        psdf = self.psdf
-        self.assertRaises(TypeError, lambda: psdf["string"] - "x")
-        self.assertRaises(TypeError, lambda: psdf["string"] - 1)
-
-        for col in self.df_cols:
-            self.assertRaises(TypeError, lambda: psdf["string"] - psdf[col])
-
-    def test_mul(self):
-        pdf, psdf = self.pdf, self.psdf
-        self.assertRaises(TypeError, lambda: psdf["string"] * "x")
-        self.assert_eq(pdf["string"] * 1, psdf["string"] * 1)
-
-        for col in self.df_cols:
-            if col in ["int", "int32"]:
-                self.assert_eq(pdf["string"] * pdf[col], psdf["string"] * psdf[col])
-            else:
-                self.assertRaises(TypeError, lambda: psdf["string"] * psdf[col])
-
-    def test_truediv(self):
-        psdf = self.psdf
-        self.assertRaises(TypeError, lambda: psdf["string"] / "x")
-        self.assertRaises(TypeError, lambda: psdf["string"] / 1)
-
-        for col in self.df_cols:
-            self.assertRaises(TypeError, lambda: psdf["string"] / psdf[col])
-
-    def test_floordiv(self):
-        psdf = self.psdf
-        self.assertRaises(TypeError, lambda: psdf["string"] // "x")
-        self.assertRaises(TypeError, lambda: psdf["string"] // 1)
-
-        for col in self.df_cols:
-            self.assertRaises(TypeError, lambda: psdf["string"] // psdf[col])
-
-    def test_mod(self):
-        psdf = self.psdf
-        self.assertRaises(TypeError, lambda: psdf["string"] % "x")
-        self.assertRaises(TypeError, lambda: psdf["string"] % 1)
-
-        for col in self.df_cols:
-            self.assertRaises(TypeError, lambda: psdf["string"] % psdf[col])
-
-    def test_pow(self):
-        psdf = self.psdf
-        self.assertRaises(TypeError, lambda: psdf["string"] ** "x")
-        self.assertRaises(TypeError, lambda: psdf["string"] ** 1)
-
-        for col in self.df_cols:
-            self.assertRaises(TypeError, lambda: psdf["string"] ** psdf[col])
-
-    def test_radd(self):
-        self.assert_eq("x" + self.pdf["string"], "x" + self.psdf["string"])
-        self.assertRaises(TypeError, lambda: 1 + self.psdf["string"])
-
-    def test_rsub(self):
-        self.assertRaises(TypeError, lambda: "x" - self.psdf["string"])
-        self.assertRaises(TypeError, lambda: 1 - self.psdf["string"])
-
-    def test_rmul(self):
-        self.assertRaises(TypeError, lambda: "x" * self.psdf["string"])
-        self.assert_eq(1 * self.pdf["string"], 1 * self.psdf["string"])
-
-    def test_rtruediv(self):
-        self.assertRaises(TypeError, lambda: "x" / self.psdf["string"])
-        self.assertRaises(TypeError, lambda: 1 / self.psdf["string"])
-
-    def test_rfloordiv(self):
-        self.assertRaises(TypeError, lambda: "x" // self.psdf["string"])
-        self.assertRaises(TypeError, lambda: 1 // self.psdf["string"])
-
-    def test_rmod(self):
-        self.assertRaises(TypeError, lambda: 1 % self.psdf["string"])
-
-    def test_rpow(self):
-        self.assertRaises(TypeError, lambda: "x" ** self.psdf["string"])
-        self.assertRaises(TypeError, lambda: 1 ** self.psdf["string"])
-
-    def test_and(self):
-        self.assertRaises(TypeError, lambda: self.psdf["string"] & True)
-        self.assertRaises(TypeError, lambda: self.psdf["string"] & False)
-        self.assertRaises(TypeError, lambda: self.psdf["string"] & self.psdf["string"])
-
-    def test_rand(self):
-        self.assertRaises(TypeError, lambda: True & self.psdf["string"])
-        self.assertRaises(TypeError, lambda: False & self.psdf["string"])
-
-    def test_or(self):
-        self.assertRaises(TypeError, lambda: self.psdf["string"] | True)
-        self.assertRaises(TypeError, lambda: self.psdf["string"] | False)
-        self.assertRaises(TypeError, lambda: self.psdf["string"] | self.psdf["string"])
-
-    def test_ror(self):
-        self.assertRaises(TypeError, lambda: True | self.psdf["string"])
-        self.assertRaises(TypeError, lambda: False | self.psdf["string"])
-
-    def test_from_to_pandas(self):
-        data = ["x", "y", "z"]
-        pser = pd.Series(data)
-        psser = ps.Series(data)
-        self.assert_eq(pser, psser._to_pandas())
-        self.assert_eq(ps.from_pandas(pser), psser)
-
-    def test_isnull(self):
-        self.assert_eq(self.pdf["string"].isnull(), self.psdf["string"].isnull())
-
-    def test_astype(self):
-        pser = pd.Series(["1", "2", "3"])
-        psser = ps.from_pandas(pser)
-        self.assert_eq(pser.astype(int), psser.astype(int))
-        self.assert_eq(pser.astype(float), psser.astype(float))
-        self.assert_eq(pser.astype(np.float32), psser.astype(np.float32))
-        self.assert_eq(pser.astype(np.int32), psser.astype(np.int32))
-        self.assert_eq(pser.astype(np.int16), psser.astype(np.int16))
-        self.assert_eq(pser.astype(np.int8), psser.astype(np.int8))
-        self.assert_eq(pser.astype(str), psser.astype(str))
-        self.assert_eq(pser.astype(bool), psser.astype(bool))
-        self.assert_eq(pser.astype("category"), psser.astype("category"))
-        cat_type = CategoricalDtype(categories=["3", "1", "2"])
-        self.assert_eq(pser.astype(cat_type), psser.astype(cat_type))
-
-    def test_neg(self):
-        self.assertRaises(TypeError, lambda: -self.psdf["string"])
-
-    def test_abs(self):
-        self.assertRaises(TypeError, lambda: abs(self.psdf["string"]))
-
-    def test_invert(self):
-        self.assertRaises(TypeError, lambda: ~self.psdf["string"])
-
-    def test_eq(self):
-        pdf, psdf = self.bool_non_numeric_pdf, self.bool_non_numeric_psdf
-        pser, psser = pdf["this"], psdf["this"]
-        other_pser, other_psser = pdf["that"], psdf["that"]
-        self.assert_eq(pser == other_pser, psser == other_psser)
-        self.assert_eq(pser == pser, psser == psser)
-
-    def test_ne(self):
-        pdf, psdf = self.bool_non_numeric_pdf, self.bool_non_numeric_psdf
-        pser, psser = pdf["this"], psdf["this"]
-        other_pser, other_psser = pdf["that"], psdf["that"]
-        self.assert_eq(pser != other_pser, psser != other_psser)
-        self.assert_eq(pser != pser, psser != psser)
-
-    def test_lt(self):
-        pdf, psdf = self.bool_non_numeric_pdf, self.bool_non_numeric_psdf
-        pser, psser = pdf["this"], psdf["this"]
-        other_pser, other_psser = pdf["that"], psdf["that"]
-        self.assert_eq(pser < other_pser, psser < other_psser)
-        self.assert_eq(pser < pser, psser < psser)
-
-    def test_le(self):
-        pdf, psdf = self.bool_non_numeric_pdf, self.bool_non_numeric_psdf
-        pser, psser = pdf["this"], psdf["this"]
-        other_pser, other_psser = pdf["that"], psdf["that"]
-        self.assert_eq(pser <= other_pser, psser <= other_psser)
-        self.assert_eq(pser <= pser, psser <= psser)
-
-    def test_gt(self):
-        pdf, psdf = self.bool_non_numeric_pdf, self.bool_non_numeric_psdf
-        pser, psser = pdf["this"], psdf["this"]
-        other_pser, other_psser = pdf["that"], psdf["that"]
-        self.assert_eq(pser > other_pser, psser > other_psser)
-        self.assert_eq(pser > pser, psser > psser)
-
-    def test_ge(self):
-        pdf, psdf = self.bool_non_numeric_pdf, self.bool_non_numeric_psdf
-        pser, psser = pdf["this"], psdf["this"]
-        other_pser, other_psser = pdf["that"], psdf["that"]
-        self.assert_eq(pser >= other_pser, psser >= other_psser)
-        self.assert_eq(pser >= pser, psser >= psser)
-
-
-class StringOpsTests(StringOpsTestsMixin, OpsTestBase):
-    pass
-
-
-@unittest.skipIf(
-    not extension_object_dtypes_available, "pandas extension object dtypes are not available"
-)
-class StringExtensionOpsTest(StringOpsTests):
-    @property
-    def pser(self):
-        return pd.Series(["x", "y", "z", None], dtype="string")
-
-    @property
-    def psser(self):
-        return ps.from_pandas(self.pser)
-
-    @property
-    def other_pser(self):
-        return pd.Series([None, "z", "y", "x"], dtype="string")
-
-    @property
-    def other_psser(self):
-        return ps.from_pandas(self.other_pser)
-
-    def test_radd(self):
-        self.assert_eq("x" + self.pser, ("x" + self.psser).astype("string"))
-        self.assertRaises(TypeError, lambda: 1 + self.psser)
-
-    def test_mul(self):
-        self.assertRaises(TypeError, lambda: self.psser * "x")
-        self.assert_eq(self.pser * 1, self.psser * 1)
-
-        with option_context("compute.ops_on_diff_frames", True):
-            for pser, psser in self.pser_psser_pairs:
-                if psser.dtype in [np.int32, np.int64]:
-                    self.assert_eq(
-                        ps.Series(["x", "yy", "zzz", None]).astype("string"),
-                        (self.psser * psser).sort_index(),
-                    )
-                else:
-                    self.assertRaises(TypeError, lambda: self.psser * psser)
-
-    def test_from_to_pandas(self):
-        data = ["x", "y", "z", None]
-        pser = pd.Series(data, dtype="string")
-        psser = ps.Series(data, dtype="string")
-        self.assert_eq(pser, psser._to_pandas())
-        self.assert_eq(ps.from_pandas(pser), psser)
-
-    def test_isnull(self):
-        self.assert_eq(self.pser.isnull(), self.psser.isnull())
-
-    def test_astype(self):
-        pser = self.pser
-        psser = self.psser
-
-        self.assert_eq(pser.astype(str).tolist(), psser.astype(str).tolist())
-
-        self.assert_eq(pser.astype("category"), psser.astype("category"))
-        cat_type = CategoricalDtype(categories=["x", "y"])
-        self.assert_eq(pser.astype(cat_type), psser.astype(cat_type))
-        for dtype in self.object_extension_dtypes:
-            if dtype in ["string", StringDtype()]:
-                self.check_extension(pser.astype(dtype), psser.astype(dtype))
-
-    def test_eq(self):
-        with option_context("compute.ops_on_diff_frames", True):
-            self.check_extension(
-                self.pser == self.other_pser, (self.psser == self.other_psser).sort_index()
-            )
-            self.check_extension(self.pser == self.pser, (self.psser == self.psser).sort_index())
-
-    def test_ne(self):
-        with option_context("compute.ops_on_diff_frames", True):
-            self.check_extension(
-                self.pser != self.other_pser, (self.psser != self.other_psser).sort_index()
-            )
-            self.check_extension(self.pser != self.pser, (self.psser != self.psser).sort_index())
-
-    def test_lt(self):
-        with option_context("compute.ops_on_diff_frames", True):
-            self.check_extension(
-                self.pser < self.other_pser, (self.psser < self.other_psser).sort_index()
-            )
-            self.check_extension(self.pser < self.pser, (self.psser < self.psser).sort_index())
-
-    def test_le(self):
-        with option_context("compute.ops_on_diff_frames", True):
-            self.check_extension(
-                self.pser <= self.other_pser, (self.psser <= self.other_psser).sort_index()
-            )
-            self.check_extension(self.pser <= self.pser, (self.psser <= self.psser).sort_index())
-
-    def test_gt(self):
-        with option_context("compute.ops_on_diff_frames", True):
-            self.check_extension(
-                self.pser > self.other_pser, (self.psser > self.other_psser).sort_index()
-            )
-            self.check_extension(self.pser > self.pser, (self.psser > self.psser).sort_index())
-
-    def test_ge(self):
-        with option_context("compute.ops_on_diff_frames", True):
-            self.check_extension(
-                self.pser >= self.other_pser, (self.psser >= self.other_psser).sort_index()
-            )
-            self.check_extension(self.pser >= self.pser, (self.psser >= self.psser).sort_index())
-
-
-if __name__ == "__main__":
-
-    from pyspark.pandas.tests.data_type_ops.test_string_ops 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/pandas/tests/data_type_ops/test_timedelta_ops.py

@@ -1,219 +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.
-#
-
-from datetime import timedelta
-
-import pandas as pd
-from pandas.api.types import CategoricalDtype
-
-import pyspark.pandas as ps
-from pyspark.pandas.tests.data_type_ops.testing_utils import OpsTestBase
-
-
-class TimedeltaOpsTestsMixin:
-    @property
-    def pser(self):
-        return pd.Series([timedelta(1), timedelta(microseconds=2), timedelta(weeks=3)])
-
-    @property
-    def psser(self):
-        return ps.from_pandas(self.pser)
-
-    @property
-    def timedelta_pdf(self):
-        psers = {
-            "this": self.pser,
-            "that": pd.Series([timedelta(0), timedelta(microseconds=1), timedelta(seconds=2)]),
-        }
-        return pd.concat(psers, axis=1)
-
-    @property
-    def timedelta_psdf(self):
-        return ps.from_pandas(self.timedelta_pdf)
-
-    @property
-    def some_timedelta(self):
-        return timedelta(weeks=2)
-
-    def test_add(self):
-        self.assertRaises(TypeError, lambda: self.psser + "x")
-        self.assertRaises(TypeError, lambda: self.psser + 1)
-
-        for psser in self.pssers:
-            self.assertRaises(TypeError, lambda: self.psser + psser)
-
-    def test_sub(self):
-        self.assertRaises(TypeError, lambda: self.psser - "x")
-        self.assertRaises(TypeError, lambda: self.psser - 1)
-
-        self.assert_eq(self.pser - self.some_timedelta, self.psser - self.some_timedelta)
-
-        pdf, psdf = self.pdf, self.psdf
-        for col in self.df_cols:
-            if col == "timedelta":
-                self.assert_eq(pdf["timedelta"] - pdf[col], psdf["timedelta"] - psdf[col])
-            else:
-                self.assertRaises(TypeError, lambda: psdf["timedelta"] - psdf[col])
-
-        pdf, psdf = self.timedelta_pdf, self.timedelta_psdf
-        self.assert_eq(pdf["that"] - pdf["this"], psdf["that"] - psdf["this"])
-
-    def test_mul(self):
-        self.assertRaises(TypeError, lambda: self.psser * "x")
-        self.assertRaises(TypeError, lambda: self.psser * 1)
-
-        for psser in self.pssers:
-            self.assertRaises(TypeError, lambda: self.psser * psser)
-
-    def test_truediv(self):
-        self.assertRaises(TypeError, lambda: self.psser / "x")
-        self.assertRaises(TypeError, lambda: self.psser / 1)
-
-        for psser in self.pssers:
-            self.assertRaises(TypeError, lambda: self.psser / psser)
-
-    def test_floordiv(self):
-        self.assertRaises(TypeError, lambda: self.psser // "x")
-        self.assertRaises(TypeError, lambda: self.psser // 1)
-
-        for psser in self.pssers:
-            self.assertRaises(TypeError, lambda: self.psser // psser)
-
-    def test_mod(self):
-        self.assertRaises(TypeError, lambda: self.psser % "x")
-        self.assertRaises(TypeError, lambda: self.psser % 1)
-
-        for psser in self.pssers:
-            self.assertRaises(TypeError, lambda: self.psser % psser)
-
-    def test_pow(self):
-        self.assertRaises(TypeError, lambda: self.psser ** "x")
-        self.assertRaises(TypeError, lambda: self.psser**1)
-
-        for psser in self.pssers:
-            self.assertRaises(TypeError, lambda: self.psser**psser)
-
-    def test_radd(self):
-        self.assertRaises(TypeError, lambda: "x" + self.psser)
-        self.assertRaises(TypeError, lambda: 1 + self.psser)
-
-    def test_rsub(self):
-        self.assertRaises(TypeError, lambda: "x" - self.psser)
-        self.assertRaises(TypeError, lambda: 1 - self.psser)
-        self.assert_eq(self.some_timedelta - self.pser, self.some_timedelta - self.psser)
-
-    def test_rmul(self):
-        self.assertRaises(TypeError, lambda: "x" * self.psser)
-        self.assertRaises(TypeError, lambda: 2 * self.psser)
-
-    def test_rtruediv(self):
-        self.assertRaises(TypeError, lambda: "x" / self.psser)
-        self.assertRaises(TypeError, lambda: 1 / self.psser)
-
-    def test_rfloordiv(self):
-        self.assertRaises(TypeError, lambda: "x" // self.psser)
-        self.assertRaises(TypeError, lambda: 1 // self.psser)
-
-    def test_rmod(self):
-        self.assertRaises(TypeError, lambda: 1 % self.psser)
-
-    def test_rpow(self):
-        self.assertRaises(TypeError, lambda: "x" ** self.psser)
-        self.assertRaises(TypeError, lambda: 1**self.psser)
-
-    def test_from_to_pandas(self):
-        data = [timedelta(1), timedelta(microseconds=2)]
-        pser = pd.Series(data)
-        psser = ps.Series(data)
-        self.assert_eq(pser, psser._to_pandas())
-        self.assert_eq(ps.from_pandas(pser), psser)
-
-    def test_isnull(self):
-        self.assert_eq(self.pser.isnull(), self.psser.isnull())
-
-    def test_astype(self):
-        pser = self.pser
-        psser = self.psser
-        target_psser = ps.Series(
-            [
-                "INTERVAL '1 00:00:00' DAY TO SECOND",
-                "INTERVAL '0 00:00:00.000002' DAY TO SECOND",
-                "INTERVAL '21 00:00:00' DAY TO SECOND",
-            ]
-        )
-        self.assert_eq(target_psser, psser.astype(str))
-        self.assert_eq(pser.astype("category"), psser.astype("category"))
-        cat_type = CategoricalDtype(categories=["a", "b", "c"])
-        self.assert_eq(pser.astype(cat_type), psser.astype(cat_type))
-
-        self.assertRaises(TypeError, lambda: psser.astype(bool))
-
-    def test_neg(self):
-        self.assertRaises(TypeError, lambda: -self.psser)
-
-    def test_abs(self):
-        self.assertRaises(TypeError, lambda: abs(self.psser))
-
-    def test_invert(self):
-        self.assertRaises(TypeError, lambda: ~self.psser)
-
-    def test_eq(self):
-        pdf, psdf = self.timedelta_pdf, self.timedelta_psdf
-        self.assert_eq(pdf["this"] == pdf["this"], psdf["this"] == psdf["this"])
-        self.assert_eq(pdf["this"] == pdf["that"], psdf["this"] == psdf["that"])
-
-    def test_ne(self):
-        pdf, psdf = self.timedelta_pdf, self.timedelta_psdf
-        self.assert_eq(pdf["this"] != pdf["this"], psdf["this"] != psdf["this"])
-        self.assert_eq(pdf["this"] != pdf["that"], psdf["this"] != psdf["that"])
-
-    def test_lt(self):
-        pdf, psdf = self.timedelta_pdf, self.timedelta_psdf
-        self.assert_eq(pdf["this"] < pdf["that"], psdf["this"] < psdf["that"])
-        self.assert_eq(pdf["this"] < pdf["this"], psdf["this"] < psdf["this"])
-
-    def test_le(self):
-        pdf, psdf = self.timedelta_pdf, self.timedelta_psdf
-        self.assert_eq(pdf["this"] <= pdf["that"], psdf["this"] <= psdf["that"])
-        self.assert_eq(pdf["this"] <= pdf["this"], psdf["this"] <= psdf["this"])
-
-    def test_gt(self):
-        pdf, psdf = self.timedelta_pdf, self.timedelta_psdf
-        self.assert_eq(pdf["this"] > pdf["that"], psdf["this"] > psdf["that"])
-        self.assert_eq(pdf["this"] > pdf["this"], psdf["this"] > psdf["this"])
-
-    def test_ge(self):
-        pdf, psdf = self.timedelta_pdf, self.timedelta_psdf
-        self.assert_eq(pdf["this"] >= pdf["that"], psdf["this"] >= psdf["that"])
-        self.assert_eq(pdf["this"] >= pdf["this"], psdf["this"] >= psdf["this"])
-
-
-class TimedeltaOpsTests(TimedeltaOpsTestsMixin, OpsTestBase):
-    pass
-
-
-if __name__ == "__main__":
-    import unittest
-    from pyspark.pandas.tests.data_type_ops.test_timedelta_ops 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)