TST: Move a consistency ewm test to test_ewm.py (#44171)

pandas-dev · Oct 25, 2021 · 82f5395 · 82f5395
1 parent 9018d32
commit 82f5395
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 9 deletions.
diff --git a/pandas/tests/window/moments/test_moments_consistency_ewm.py b/pandas/tests/window/moments/test_moments_consistency_ewm.py
@@ -9,15 +9,6 @@
 import pandas._testing as tm
 
 
-@pytest.mark.parametrize("func", ["cov", "corr"])
-def test_ewm_pairwise_cov_corr(func, frame):
-    result = getattr(frame.ewm(span=10, min_periods=5), func)()
-    result = result.loc[(slice(None), 1), 5]
-    result.index = result.index.droplevel(1)
-    expected = getattr(frame[1].ewm(span=10, min_periods=5), func)(frame[5])
-    tm.assert_series_equal(result, expected, check_names=False)
-
-
 def create_mock_weights(obj, com, adjust, ignore_na):
     if isinstance(obj, DataFrame):
         if not len(obj.columns):

diff --git a/pandas/tests/window/test_ewm.py b/pandas/tests/window/test_ewm.py
@@ -657,3 +657,12 @@ def test_ewm_alpha_arg(series):
         s.ewm(span=10.0, alpha=0.5)
     with pytest.raises(ValueError, match=msg):
         s.ewm(halflife=10.0, alpha=0.5)
+
+
+@pytest.mark.parametrize("func", ["cov", "corr"])
+def test_ewm_pairwise_cov_corr(func, frame):
+    result = getattr(frame.ewm(span=10, min_periods=5), func)()
+    result = result.loc[(slice(None), 1), 5]
+    result.index = result.index.droplevel(1)
+    expected = getattr(frame[1].ewm(span=10, min_periods=5), func)(frame[5])
+    tm.assert_series_equal(result, expected, check_names=False)