Fix.

databricks · Dec 22, 2020 · 87d0a2c · 87d0a2c
1 parent 4c779d0
commit 87d0a2c
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/databricks/koalas/tests/test_stats.py b/databricks/koalas/tests/test_stats.py
@@ -94,11 +94,11 @@ def test_sum(self):
     def test_abs(self):
         pdf = pd.DataFrame(
             {
-                "A": [1, -2, 3, -4, 5],
-                "B": [1.0, -2, 3, -4, 5],
-                "C": [-6.0, -7, -8, -9, 10],
-                "D": ["a", "b", "c", "d", "e"],
-                "E": [True, False, False, True, True],
+                "A": [1, -2, np.nan, -4, 5],
+                "B": [1.0, -2, np.nan, -4, 5],
+                "C": [-6.0, -7, -8, np.nan, 10],
+                "D": ["a", "b", "c", "d", np.nan],
+                "E": [True, np.nan, False, True, True],
             }
         )
         kdf = ks.from_pandas(pdf)