enable in test

Author: xinrong-meng

python/pyspark/pandas/tests/computation/test_binary_ops.py

@@ -112,69 +112,70 @@ def test_binary_operator_sub(self):
         self.assertRaisesRegex(TypeError, ks_err_msg, lambda: psdf["a"] - psdf["b"])
 
     def test_divide_by_zero_behavior(self):
-        # float / float
-        for dtype in [np.float32, np.float64]:
+        with ps.option_context("compute.ansi_mode_support", True):
+            # float / float
+            for dtype in [np.float32, np.float64]:
+                pdf = pd.DataFrame(
+                    {
+                        "a": [1.0, -1.0, 0.0, np.nan],
+                        "b": [0.0, 0.0, 0.0, 0.0],
+                    },
+                    dtype=dtype,
+                )
+                psdf = ps.from_pandas(pdf)
+
+                self.assert_eq(psdf["a"] / psdf["b"], pdf["a"] / pdf["b"])
+
+            # int / int
+            for dtype in [np.int32, np.int64]:
+                pdf = pd.DataFrame(
+                    {
+                        "a": [1, -1, 0],
+                        "b": [0, 0, 0],
+                    },
+                    dtype=dtype,
+                )
+                psdf = ps.from_pandas(pdf)
+                self.assert_eq(psdf["a"] / psdf["b"], pdf["a"] / pdf["b"])
+
+            # float / int
             pdf = pd.DataFrame(
                 {
-                    "a": [1.0, -1.0, 0.0, np.nan],
-                    "b": [0.0, 0.0, 0.0, 0.0],
-                },
-                dtype=dtype,
+                    "a": pd.Series([1.0, -1.0, 0.0, np.nan]),
+                    "b": pd.Series([0, 0, 0, 0]),
+                }
             )
             psdf = ps.from_pandas(pdf)
-
             self.assert_eq(psdf["a"] / psdf["b"], pdf["a"] / pdf["b"])
 
-        # int / int
-        for dtype in [np.int32, np.int64]:
+            # int / float
             pdf = pd.DataFrame(
                 {
-                    "a": [1, -1, 0],
-                    "b": [0, 0, 0],
-                },
-                dtype=dtype,
+                    "a": pd.Series([1, -1, 0]),
+                    "b": pd.Series([0.0, 0.0, 0.0]),
+                }
             )
             psdf = ps.from_pandas(pdf)
             self.assert_eq(psdf["a"] / psdf["b"], pdf["a"] / pdf["b"])
 
-        # float / int
-        pdf = pd.DataFrame(
-            {
-                "a": pd.Series([1.0, -1.0, 0.0, np.nan]),
-                "b": pd.Series([0, 0, 0, 0]),
-            }
-        )
-        psdf = ps.from_pandas(pdf)
-        self.assert_eq(psdf["a"] / psdf["b"], pdf["a"] / pdf["b"])
-
-        # int / float
-        pdf = pd.DataFrame(
-            {
-                "a": pd.Series([1, -1, 0]),
-                "b": pd.Series([0.0, 0.0, 0.0]),
-            }
-        )
-        psdf = ps.from_pandas(pdf)
-        self.assert_eq(psdf["a"] / psdf["b"], pdf["a"] / pdf["b"])
-
-        # bool
-        pdf = pd.DataFrame(
-            {
-                "a": pd.Series([True, False]),
-                "b": pd.Series([0, 0]),
-            }
-        )
-        psdf = ps.from_pandas(pdf)
-        self.assert_eq(psdf["a"] / psdf["b"], pdf["a"] / pdf["b"])
+            # bool
+            pdf = pd.DataFrame(
+                {
+                    "a": pd.Series([True, False]),
+                    "b": pd.Series([0, 0]),
+                }
+            )
+            psdf = ps.from_pandas(pdf)
+            self.assert_eq(psdf["a"] / psdf["b"], pdf["a"] / pdf["b"])
 
-        pdf = pd.DataFrame(
-            {
-                "a": pd.Series([True, False]),
-                "b": pd.Series([0.0, 0.0]),
-            }
-        )
-        psdf = ps.from_pandas(pdf)
-        self.assert_eq(psdf["a"] / psdf["b"], pdf["a"] / pdf["b"])
+            pdf = pd.DataFrame(
+                {
+                    "a": pd.Series([True, False]),
+                    "b": pd.Series([0.0, 0.0]),
+                }
+            )
+            psdf = ps.from_pandas(pdf)
+            self.assert_eq(psdf["a"] / psdf["b"], pdf["a"] / pdf["b"])
 
     def test_binary_operator_truediv(self):
         # Positive