@@ -858,18 +858,18 @@ def test_modulo_zero_int(self):
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 = np .int64 )
861
- ea_array = pd . array ([1 , 2 , 3 , 4 , 5 ], dtype = "Int64" )
861
+ ea_array = array ([1 , 2 , 3 , 4 , 5 ], dtype = "Int64" )
862
862
result = np_array * ea_array
863
863
assert isinstance (result , type (ea_array ))
864
864
tm .assert_equal (
865
- result , pd . array ([1 , 4 , 9 , 16 , 25 ], dtype = "Int64" )
865
+ result , array ([1 , 4 , 9 , 16 , 25 ], dtype = "Int64" )
866
866
)
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
870
NP_array = np .array (
871
871
list (range (10 )), dtype = np .int64 ).reshape (10 , 1 )
872
- EA_array = pd . array (list (range (10 )), dtype = "Int64" ).reshape (10 , 1 )
872
+ EA_array = array (list (range (10 )), dtype = "Int64" ).reshape (10 , 1 )
873
873
result_np = df * NP_array
874
874
assert isinstance (result_np , np .ndarray )
875
875
tm .assert_equal (result_np .shape , (10 , 5 ))
@@ -878,7 +878,7 @@ def test_df_mul_np_and_ea_array_shape_and_errors(self):
878
878
_ = df * EA_array
879
879
880
880
def test_non_1d_ea_raises_typeerror (self ):
881
- ea_array = pd . array ([1 , 2 , 3 , 4 , 5 ], dtype = "Int64" ).reshape (5 , 1 )
881
+ ea_array = array ([1 , 2 , 3 , 4 , 5 ], dtype = "Int64" ).reshape (5 , 1 )
882
882
np_array = np .array ([1 , 2 , 3 , 4 , 5 ], dtype = np .int64 ).reshape (5 , 1 )
883
883
884
884
with pytest .raises (NotImplementedError ):
0 commit comments