base.py 742 B

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