@@ -858,18 +858,18 @@ def test_modulo_zero_int(self):
858858
859859 def test_np_array_mul_ea_array_returns_extensionarray (self ):
860860 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" )
862862 result = np_array * ea_array
863863 assert isinstance (result , type (ea_array ))
864864 tm .assert_equal (
865- result , pd . array ([1 , 4 , 9 , 16 , 25 ], dtype = "Int64" )
865+ result , array ([1 , 4 , 9 , 16 , 25 ], dtype = "Int64" )
866866 )
867867
868868 def test_df_mul_np_and_ea_array_shape_and_errors (self ):
869869 df = pd .DataFrame (np .arange (50 ).reshape (10 , 5 )).notna ().values
870870 NP_array = np .array (
871871 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 )
873873 result_np = df * NP_array
874874 assert isinstance (result_np , np .ndarray )
875875 tm .assert_equal (result_np .shape , (10 , 5 ))
@@ -878,7 +878,7 @@ def test_df_mul_np_and_ea_array_shape_and_errors(self):
878878 _ = df * EA_array
879879
880880 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 )
882882 np_array = np .array ([1 , 2 , 3 , 4 , 5 ], dtype = np .int64 ).reshape (5 , 1 )
883883
884884 with pytest .raises (NotImplementedError ):
0 commit comments