Skip to content

Commit ae5b4c1

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent f0c9157 commit ae5b4c1

File tree

1 file changed

+13
-5
lines changed

1 file changed

+13
-5
lines changed

pandas/tests/arithmetic/test_numeric.py

Lines changed: 13 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -855,7 +855,7 @@ def test_modulo_zero_int(self):
855855
result = 0 % s
856856
expected = Series([np.nan, 0.0])
857857
tm.assert_series_equal(result, expected)
858-
858+
859859
def test_np_array_mul_ea_array_returns_extensionarray(self):
860860
np_array = np.array([1, 2, 3, 4, 5], dtype=tm.SIGNED_INT_NUMPY_DTYPES[0])
861861
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):
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
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+
)
872876
result_np = df * NP_array
873877
tm.assert_isinstance(result_np, np.ndarray)
874878
tm.assert_equal(result_np.shape, (10, 5))
@@ -877,8 +881,12 @@ def test_df_mul_np_and_ea_array_shape_and_errors(self):
877881
_ = df * EA_array
878882

879883
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)
882890

883891
with tm.assert_raises(TypeError):
884892
_ = ea_array * np_array

0 commit comments

Comments
 (0)