@@ -811,7 +811,7 @@ def assert_categorical_equal(
811
811
right : Categorical ,
812
812
check_dtype : bool = True ,
813
813
check_category_order : bool = True ,
814
- obj : str = "Categorical"
814
+ obj : str = "Categorical" ,
815
815
) -> None :
816
816
"""Test that Categoricals are equivalent.
817
817
@@ -858,10 +858,10 @@ def assert_categorical_equal(
858
858
859
859
860
860
def assert_interval_array_equal (
861
- left : IntervalArray ,
862
- right : IntervalArray ,
863
- exact : str = "equiv" ,
864
- obj : str = "IntervalArray"
861
+ left : IntervalArray ,
862
+ right : IntervalArray ,
863
+ exact : str = "equiv" ,
864
+ obj : str = "IntervalArray" ,
865
865
) -> None :
866
866
"""Test that two IntervalArrays are equivalent.
867
867
@@ -889,9 +889,7 @@ def assert_interval_array_equal(
889
889
890
890
891
891
def assert_period_array_equal (
892
- left : PeriodArray ,
893
- right : PeriodArray ,
894
- obj : str = "PeriodArray"
892
+ left : PeriodArray , right : PeriodArray , obj : str = "PeriodArray"
895
893
) -> None :
896
894
_check_isinstance (left , right , PeriodArray )
897
895
@@ -902,9 +900,7 @@ def assert_period_array_equal(
902
900
903
901
904
902
def assert_datetime_array_equal (
905
- left : DatetimeArray ,
906
- right : DatetimeArray ,
907
- obj : str = "DatetimeArray"
903
+ left : DatetimeArray , right : DatetimeArray , obj : str = "DatetimeArray"
908
904
) -> None :
909
905
__tracebackhide__ = True
910
906
_check_isinstance (left , right , DatetimeArray )
@@ -915,9 +911,7 @@ def assert_datetime_array_equal(
915
911
916
912
917
913
def assert_timedelta_array_equal (
918
- left : TimedeltaArray ,
919
- right : TimedeltaArray ,
920
- obj : str = "TimedeltaArray"
914
+ left : TimedeltaArray , right : TimedeltaArray , obj : str = "TimedeltaArray"
921
915
) -> None :
922
916
__tracebackhide__ = True
923
917
_check_isinstance (left , right , TimedeltaArray )
@@ -960,7 +954,7 @@ def assert_numpy_array_equal(
960
954
err_msg : Optional [str ] = None ,
961
955
check_same : Optional [str ] = None ,
962
956
obj : str = "numpy array" ,
963
- ):
957
+ ) -> None :
964
958
""" Checks that 'np.ndarray' is equivalent
965
959
966
960
Parameters
@@ -1431,9 +1425,9 @@ def assert_frame_equal(
1431
1425
1432
1426
1433
1427
def assert_equal (
1434
- left : Union [DataFrame , AnyArrayLike ],
1435
- right : Union [DataFrame , AnyArrayLike ],
1436
- ** kwargs
1428
+ left : Union [DataFrame , AnyArrayLike ],
1429
+ right : Union [DataFrame , AnyArrayLike ],
1430
+ ** kwargs
1437
1431
) -> None :
1438
1432
"""
1439
1433
Wrapper for tm.assert_*_equal to dispatch to the appropriate test function.
0 commit comments