TST: Sparse where test

Author: Licht-T

pandas/tests/sparse/test_frame.py

@@ -1385,3 +1385,38 @@ def test_numpy_func_call(self):
                  'std', 'min', 'max']
         for func in funcs:
             getattr(np, func)(self.frame)
+
+    def test_where(self):
+        data = [[1, 2], [3, 4]]
+
+        sparse_df = SparseDataFrame(data)
+        result = sparse_df.where(sparse_df >= 2)
+
+        dense_df = DataFrame(data)
+        expected = dense_df.where(dense_df >= 2)
+
+        tm.assert_frame_equal(result, expected)
+
+    def test_where_with_zero_other(self):
+        data = [[1, 2], [3, 4]]
+        other = 0
+
+        sparse_df = SparseDataFrame(data)
+        result = sparse_df.where(sparse_df >= 2, other)
+
+        dense_df = DataFrame(data)
+        expected = dense_df.where(dense_df >= 2, other)
+
+        tm.assert_frame_equal(result, expected)
+
+    def test_where_with_float_other(self):
+        data = [[1, 2], [3, 4]]
+        other = 0.1
+
+        sparse_df = SparseDataFrame(data)
+        result = sparse_df.where(sparse_df >= 2, other)
+
+        dense_df = DataFrame(data)
+        expected = dense_df.where(dense_df >= 2, other)
+
+        tm.assert_frame_equal(result, expected)

pandas/tests/sparse/test_series.py

@@ -1384,6 +1384,41 @@ def test_numpy_func_call(self):
             for series in ('bseries', 'zbseries'):
                 getattr(np, func)(getattr(self, series))
 
+    def test_where(self):
+        data = [1, 2, 3, 4]
+
+        sparse = SparseSeries(data)
+        result = sparse.where(sparse >= 2)
+
+        dense = Series(data)
+        expected = dense.where(dense >= 2)
+
+        tm.assert_series_equal(result, expected)
+
+    def test_where_with_zero_other(self):
+        data = [1, 2, 3, 4]
+        other = 0
+
+        sparse = SparseSeries(data)
+        result = sparse.where(sparse >= 2, other)
+
+        dense = Series(data)
+        expected = dense.where(dense >= 2, other)
+
+        tm.assert_series_equal(result, expected)
+
+    def test_where_with_float_other(self):
+        data = [1, 2, 3, 4]
+        other = 0.1
+
+        sparse = SparseSeries(data)
+        result = sparse.where(sparse >= 2, other)
+
+        dense = Series(data)
+        expected = dense.where(dense >= 2, other)
+
+        tm.assert_series_equal(result, expected)
+
 
 @pytest.mark.parametrize(
     'datetime_type', (np.datetime64,