Add tests for Series/DataFrame.to_sparse

pandas-dev · jreback · Jun 19, 2019 · Jun 5, 2019 · Jun 5, 2019 · Jun 5, 2019
commit eede9b8b830165f281b51dfb45aae49f9a706db2
diff --git a/pandas/tests/frame/test_deprecations.py b/pandas/tests/frame/test_deprecations.py
@@ -0,0 +1,20 @@
+import numpy as np
+from numpy import nan
+import pytest
+
+import pandas as pd
+from pandas import DataFrame
+from pandas.core.sparse import frame as spf
+from pandas.util import testing as tm
+
+
+@pytest.mark.filterwarnings("ignore:Sparse:FutureWarning")
+def test_deprecated_to_sparse():
+    df = pd.DataFrame({"A": [1, np.nan, 3]})
+    sparse_df = pd.SparseDataFrame({"A": [1, np.nan, 3]})
+
+    # Deprecated 0.25.0
+    with tm.assert_produces_warning(FutureWarning,
+                                    check_stacklevel=False):
+            result = df.to_sparse()
+    tm.assert_frame_equal(result, sparse_df)
diff --git a/pandas/tests/groupby/aggregate/test_other.py b/pandas/tests/groupby/aggregate/test_other.py
@@ -385,6 +385,7 @@ def __call__(self, x):
         tm.assert_frame_equal(result, expected)
 
 
+@pytest.mark.filterwarnings("ignore:NDFrame:FutureWarning")
 def test_agg_over_numpy_arrays():
     # GH 3788
     df = pd.DataFrame([[1, np.array([10, 20, 30])],

diff --git a/pandas/tests/series/test_deprecations.py b/pandas/tests/series/test_deprecations.py
@@ -0,0 +1,20 @@
+import numpy as np
+from numpy import nan
+import pytest
+
+import pandas as pd
+from pandas import Series
+from pandas.core.sparse import series as sps
+from pandas.util import testing as tm
+
+
+@pytest.mark.filterwarnings("ignore:Sparse:FutureWarning")
+def test_deprecated_to_sparse():
+    ser = Series([1, np.nan, 3])
+    sparse_ser = pd.SparseSeries([1, np.nan, 3])
+
+    # Deprecated 0.25.0
+    with tm.assert_produces_warning(FutureWarning,
+                                    check_stacklevel=False):
+        result = ser.to_sparse()
+    tm.assert_series_equal(result, sparse_ser)
diff --git a/pandas/tests/sparse/frame/test_frame.py b/pandas/tests/sparse/frame/test_frame.py
@@ -1294,15 +1294,17 @@ def test_default_fill_value_with_no_data(self):
                                     default_fill_value=1.0)
         tm.assert_frame_equal(expected, result)
 
-    def test_deprecated_to_dense(self):
-        df = pd.DataFrame({"A": [1, np.nan, 3]})
-        sparse_df = pd.SparseDataFrame({"A": [1, np.nan, 3]})
 
-        # Deprecated 0.25.0
-        with tm.assert_produces_warning(FutureWarning,
-                                        check_stacklevel=False):
-            result = sparse_df.to_dense()
-        tm.assert_frame_equal(result, df)
+#@pytest.mark.filterwarnings("ignore:Series:FutureWarning")
+def test_deprecated_to_dense():
+    df = pd.DataFrame({"A": [1, np.nan, 3]})
+
+    # Deprecated 0.25.0
+    with tm.assert_produces_warning(FutureWarning,
+                                    check_stacklevel=False):
+        sparse_df = pd.SparseDataFrame({"A": [1, np.nan, 3]})
+        result = sparse_df.to_dense()
+    tm.assert_frame_equal(result, df)
 
 
 @pytest.mark.filterwarnings("ignore:Sparse:FutureWarning")

diff --git a/pandas/tests/sparse/series/test_series.py b/pandas/tests/sparse/series/test_series.py
@@ -1041,15 +1041,6 @@ def test_memory_usage_deep(self, deep, fill_value):
 
         assert sparse_usage < dense_usage
 
-    def test_deprecate_to_dense(self):
-        ser = pd.Series([1, 2, 3])
-        sparse_ser = pd.SparseSeries([1, 2, 3])
-
-        # Deprecated 0.25.0
-        with tm.assert_produces_warning(FutureWarning):
-            result = sparse_ser.to_dense()
-        tm.assert_series_equal(result, ser)
-
 
 @pytest.mark.filterwarnings("ignore:Sparse:FutureWarning")
 @pytest.mark.filterwarnings("ignore:Series:FutureWarning")
@@ -1570,3 +1561,14 @@ def test_block_deprecated():
     s = SparseSeries([1])
     with tm.assert_produces_warning(FutureWarning):
         s.block
+
+
+@pytest.mark.filterwarnings("ignore:Sparse:FutureWarning")
+def test_deprecate_to_dense():
+    ser = pd.Series([1, 2, 3])
+    sparse_ser = pd.SparseSeries([1, 2, 3])
+
+    # Deprecated 0.25.0
+    with tm.assert_produces_warning(FutureWarning):
+        result = sparse_ser.to_dense()
+        tm.assert_series_equal(result, ser)
diff --git a/pandas/tests/test_multilevel.py b/pandas/tests/test_multilevel.py
@@ -55,6 +55,7 @@ def setup_method(self, method):
         self.ymd.index.set_names(['year', 'month', 'day'], inplace=True)
 
 
+@pytest.mark.filterwarnings("ignore:NDFrame:FutureWarning")
 class TestMultiLevel(Base):
 
     def test_append(self):