@@ -855,7 +855,7 @@ def test_modulo_zero_int(self):
855
855
result = 0 % s
856
856
expected = Series ([np .nan , 0.0 ])
857
857
tm .assert_series_equal (result , expected )
858
-
858
+
859
859
def test_np_array_mul_ea_array_returns_extensionarray (self ):
860
860
np_array = np .array ([1 , 2 , 3 , 4 , 5 ], dtype = tm .SIGNED_INT_NUMPY_DTYPES [0 ])
861
861
ea_array = pd .array ([1 , 2 , 3 , 4 , 5 ], dtype = tm .SIGNED_INT_EA_DTYPES [0 ])
@@ -867,8 +867,12 @@ def test_np_array_mul_ea_array_returns_extensionarray(self):
867
867
868
868
def test_df_mul_np_and_ea_array_shape_and_errors (self ):
869
869
df = pd .DataFrame (np .arange (50 ).reshape (10 , 5 )).notna ().values
870
- NP_array = pd .array (list (range (10 )), dtype = tm .SIGNED_INT_NUMPY_DTYPES [0 ]).reshape (10 , 1 )
871
- EA_array = pd .array (list (range (10 )), dtype = tm .SIGNED_INT_EA_DTYPES [0 ]).reshape (10 , 1 )
870
+ NP_array = pd .array (
871
+ list (range (10 )), dtype = tm .SIGNED_INT_NUMPY_DTYPES [0 ]
872
+ ).reshape (10 , 1 )
873
+ EA_array = pd .array (list (range (10 )), dtype = tm .SIGNED_INT_EA_DTYPES [0 ]).reshape (
874
+ 10 , 1
875
+ )
872
876
result_np = df * NP_array
873
877
tm .assert_isinstance (result_np , np .ndarray )
874
878
tm .assert_equal (result_np .shape , (10 , 5 ))
@@ -877,8 +881,12 @@ def test_df_mul_np_and_ea_array_shape_and_errors(self):
877
881
_ = df * EA_array
878
882
879
883
def test_non_1d_ea_raises_typeerror (self ):
880
- ea_array = pd .array ([1 , 2 , 3 , 4 , 5 ], dtype = tm .SIGNED_INT_EA_DTYPES [0 ]).reshape (5 , 1 )
881
- np_array = np .array ([1 , 2 , 3 , 4 , 5 ], dtype = tm .SIGNED_INT_NUMPY_DTYPES [0 ]).reshape (5 , 1 )
884
+ ea_array = pd .array ([1 , 2 , 3 , 4 , 5 ], dtype = tm .SIGNED_INT_EA_DTYPES [0 ]).reshape (
885
+ 5 , 1
886
+ )
887
+ np_array = np .array (
888
+ [1 , 2 , 3 , 4 , 5 ], dtype = tm .SIGNED_INT_NUMPY_DTYPES [0 ]
889
+ ).reshape (5 , 1 )
882
890
883
891
with tm .assert_raises (TypeError ):
884
892
_ = ea_array * np_array
0 commit comments