@@ -615,16 +615,16 @@ def test_to_integer_array_float():
615
615
616
616
617
617
@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 )
628
628
tm .assert_extension_array_equal (result , expected )
629
629
630
630
0 commit comments