@@ -79,7 +79,8 @@ class BaseJSON:
79
79
# The default assert_series_equal eventually does a
80
80
# Series.values, which raises. We work around it by
81
81
# converting the UserDicts to dicts.
82
- def assert_series_equal (self , left , right , ** kwargs ):
82
+ @classmethod
83
+ def assert_series_equal (cls , left , right , * args , ** kwargs ):
83
84
if left .dtype .name == "json" :
84
85
assert left .dtype == right .dtype
85
86
left = pd .Series (
@@ -90,9 +91,10 @@ def assert_series_equal(self, left, right, **kwargs):
90
91
index = right .index ,
91
92
name = right .name ,
92
93
)
93
- tm .assert_series_equal (left , right , ** kwargs )
94
+ tm .assert_series_equal (left , right , * args , * *kwargs )
94
95
95
- def assert_frame_equal (self , left , right , * args , ** kwargs ):
96
+ @classmethod
97
+ def assert_frame_equal (cls , left , right , * args , ** kwargs ):
96
98
obj_type = kwargs .get ("obj" , "DataFrame" )
97
99
tm .assert_index_equal (
98
100
left .columns ,
@@ -107,7 +109,7 @@ def assert_frame_equal(self, left, right, *args, **kwargs):
107
109
jsons = (left .dtypes == "json" ).index
108
110
109
111
for col in jsons :
110
- self .assert_series_equal (left [col ], right [col ], * args , ** kwargs )
112
+ cls .assert_series_equal (left [col ], right [col ], * args , ** kwargs )
111
113
112
114
left = left .drop (columns = jsons )
113
115
right = right .drop (columns = jsons )
0 commit comments