From e2c87576a8e334c6b9e9b5fda5d2f405cc444c16 Mon Sep 17 00:00:00 2001 From: Matthew Roeschke <10647082+mroeschke@users.noreply.github.com> Date: Wed, 18 Oct 2023 17:27:31 -0700 Subject: [PATCH] TST: Test patching over fake instead of real method in accessor test --- pandas/tests/test_register_accessor.py | 18 ++++++------------ 1 file changed, 6 insertions(+), 12 deletions(-) diff --git a/pandas/tests/test_register_accessor.py b/pandas/tests/test_register_accessor.py index 5b200711f4b36..4e569dc40005d 100644 --- a/pandas/tests/test_register_accessor.py +++ b/pandas/tests/test_register_accessor.py @@ -82,19 +82,13 @@ def test_accessor_works(): def test_overwrite_warns(): - # Need to restore mean - mean = pd.Series.mean - try: - with tm.assert_produces_warning(UserWarning) as w: - pd.api.extensions.register_series_accessor("mean")(MyAccessor) + match = r".*MyAccessor.*fake.*Series.*" + with tm.assert_produces_warning(UserWarning, match=match): + with ensure_removed(pd.Series, "fake"): + setattr(pd.Series, "fake", 123) + pd.api.extensions.register_series_accessor("fake")(MyAccessor) s = pd.Series([1, 2]) - assert s.mean.prop == "item" - msg = str(w[0].message) - assert "mean" in msg - assert "MyAccessor" in msg - assert "Series" in msg - finally: - pd.Series.mean = mean + assert s.fake.prop == "item" def test_raises_attribute_error():