From ef9dc41d28b7ca2e0caa597c7ad523c9884d15e0 Mon Sep 17 00:00:00 2001 From: Matthew Roeschke Date: Sun, 24 Oct 2021 15:06:49 -0700 Subject: [PATCH] TST: Move a consistency ewm test to test_ewm.py --- .../tests/window/moments/test_moments_consistency_ewm.py | 9 --------- pandas/tests/window/test_ewm.py | 9 +++++++++ 2 files changed, 9 insertions(+), 9 deletions(-) diff --git a/pandas/tests/window/moments/test_moments_consistency_ewm.py b/pandas/tests/window/moments/test_moments_consistency_ewm.py index b41d2ec23a52d..800ee2164693b 100644 --- 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 index 21c0099bbc0e6..23c3a0ef27fef 100644 --- 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)