overwrite tests for sparse array

Author: thoo

pandas/tests/extension/base/methods.py

@@ -111,25 +111,19 @@ def test_fillna_copy_frame(self, data_missing):
 
         filled_val = df.iloc[0, 0]
         result = df.fillna(filled_val)
-        assert df.values.base is not result.values.base
 
-        if isinstance(arr, pd.SparseArray):
-            assert df.A._values.to_dense() is arr.to_dense()
-        else:
-            assert df.A._values is arr
+        assert df.values.base is not result.values.base
+        assert df.A._values is arr
 
     def test_fillna_copy_series(self, data_missing):
         arr = data_missing.take([1, 1])
         ser = pd.Series(arr)
 
         filled_val = ser[0]
         result = ser.fillna(filled_val)
-        assert ser._values is not result._values
 
-        if isinstance(arr, pd.SparseArray):
-            assert ser._values.to_dense() is arr.to_dense()
-        else:
-            assert ser._values is arr
+        assert ser._values is not result._values
+        assert ser._values is arr
 
     def test_fillna_length_mismatch(self, data_missing):
         with (tm.assert_raises_regex(ValueError,

pandas/tests/extension/test_sparse.py

@@ -231,6 +231,26 @@ def test_combine_le(self, data_repeated):
         ], fill_value=False))
         self.assert_series_equal(result, expected)
 
+    def test_fillna_copy_frame(self, data_missing):
+        arr = data_missing.take([1, 1])
+        df = pd.DataFrame({"A": arr})
+
+        filled_val = df.iloc[0, 0]
+        result = df.fillna(filled_val)
+
+        assert df.values.base is not result.values.base
+        assert df.A._values.to_dense() is arr.to_dense()
+
+    def test_fillna_copy_series(self, data_missing):
+        arr = data_missing.take([1, 1])
+        ser = pd.Series(arr)
+
+        filled_val = ser[0]
+        result = ser.fillna(filled_val)
+
+        assert ser._values is not result._values
+        assert ser._values.to_dense() is arr.to_dense()
+
     @pytest.mark.skip(reason="Not Applicable")
     def test_fillna_length_mismatch(self, data_missing):
         pass