zmc
2023-10-12 ed135d79df12a2466b52dae1a82326941211dcc9
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
import pandas._testing as tm
 
 
class BaseExtensionTests:
    # classmethod and different signature is needed
    # to make inheritance compliant with mypy
    @classmethod
    def assert_equal(cls, left, right, **kwargs):
        return tm.assert_equal(left, right, **kwargs)
 
    @classmethod
    def assert_series_equal(cls, left, right, *args, **kwargs):
        return tm.assert_series_equal(left, right, *args, **kwargs)
 
    @classmethod
    def assert_frame_equal(cls, left, right, *args, **kwargs):
        return tm.assert_frame_equal(left, right, *args, **kwargs)
 
    @classmethod
    def assert_extension_array_equal(cls, left, right, *args, **kwargs):
        return tm.assert_extension_array_equal(left, right, *args, **kwargs)