Skip to content

Commit 775114a

Browse files
committed
fix tests
1 parent c21412c commit 775114a

File tree

1 file changed

+10
-10
lines changed

1 file changed

+10
-10
lines changed

pandas/tests/arrays/test_integer.py

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -615,16 +615,16 @@ def test_to_integer_array_float():
615615

616616

617617
@pytest.mark.parametrize(
618-
'bool_values, int_values, result_dtype',
619-
[([False, True], [0, 1], Int64Dtype()),
620-
([False, True], [0, 1], 'Int64'),
621-
([False, True, np.nan], [0, 1, np.nan], Int64Dtype()),
622-
([False, True], [0, 1], None),
623-
([False, True, np.nan], [0, 1, np.nan], None)])
624-
def test_to_integer_array_bool(bool_values, int_values, result_dtype):
625-
result = integer_array(bool_values, dtype=result_dtype)
626-
assert result.dtype == result_dtype
627-
expected = integer_array(int_values, dtype=result_dtype)
618+
'bool_values, int_values, target_dtype, expected_dtype',
619+
[([False, True], [0, 1], Int64Dtype(), Int64Dtype()),
620+
([False, True], [0, 1], 'Int64', Int64Dtype()),
621+
([False, True, np.nan], [0, 1, np.nan], Int64Dtype(), Int64Dtype()),
622+
([False, True], [0, 1], None, Int64Dtype()),
623+
([False, True, np.nan], [0, 1, np.nan], None, Int64Dtype())])
624+
def test_to_integer_array_bool(bool_values, int_values, target_dtype, expected_dtype):
625+
result = integer_array(bool_values, dtype=target_dtype)
626+
assert result.dtype == expected_dtype
627+
expected = integer_array(int_values, dtype=target_dtype)
628628
tm.assert_extension_array_equal(result, expected)
629629

630630

0 commit comments

Comments
 (0)