File tree 1 file changed +6
-12
lines changed
1 file changed +6
-12
lines changed Original file line number Diff line number Diff line change @@ -82,19 +82,13 @@ def test_accessor_works():
82
82
83
83
84
84
def test_overwrite_warns ():
85
- # Need to restore mean
86
- mean = pd . Series . mean
87
- try :
88
- with tm . assert_produces_warning ( UserWarning ) as w :
89
- pd .api .extensions .register_series_accessor ("mean " )(MyAccessor )
85
+ match = r".*MyAccessor.*fake.*Series.*"
86
+ with tm . assert_produces_warning ( UserWarning , match = match ):
87
+ with ensure_removed ( pd . Series , "fake" ) :
88
+ setattr ( pd . Series , "fake" , 123 )
89
+ pd .api .extensions .register_series_accessor ("fake " )(MyAccessor )
90
90
s = pd .Series ([1 , 2 ])
91
- assert s .mean .prop == "item"
92
- msg = str (w [0 ].message )
93
- assert "mean" in msg
94
- assert "MyAccessor" in msg
95
- assert "Series" in msg
96
- finally :
97
- pd .Series .mean = mean
91
+ assert s .fake .prop == "item"
98
92
99
93
100
94
def test_raises_attribute_error ():
You can’t perform that action at this time.
0 commit comments