@@ -133,24 +133,37 @@ def test_to_boolean_array_error(values):
133
133
pd .array (values , dtype = "boolean" )
134
134
135
135
136
- def test_to_boolean_array_integer_like ():
137
- # integers of 0's and 1's
138
- result = pd .array ([1 , 0 , 1 , 0 ], dtype = "boolean" )
136
+ def test_to_boolean_array_from_integer_array ():
137
+ result = pd .array (np .array ([1 , 0 , 1 , 0 ]), dtype = "boolean" )
139
138
expected = pd .array ([True , False , True , False ], dtype = "boolean" )
140
139
tm .assert_extension_array_equal (result , expected )
141
140
142
- result = pd .array (np .array ([1 , 0 , 1 , 0 ]), dtype = "boolean" )
141
+ # with missing values
142
+ result = pd .array (np .array ([1 , 0 , 1 , None ]), dtype = "boolean" )
143
+ expected = pd .array ([True , False , True , None ], dtype = "boolean" )
143
144
tm .assert_extension_array_equal (result , expected )
144
145
146
+
147
+ def test_to_boolean_array_from_float_array ():
145
148
result = pd .array (np .array ([1.0 , 0.0 , 1.0 , 0.0 ]), dtype = "boolean" )
149
+ expected = pd .array ([True , False , True , False ], dtype = "boolean" )
146
150
tm .assert_extension_array_equal (result , expected )
147
151
148
152
# with missing values
149
- result = pd .array ([ 1 , 0 , 1 , None ] , dtype = "boolean" )
153
+ result = pd .array (np . array ([ 1.0 , 0.0 , 1.0 , np . nan ]) , dtype = "boolean" )
150
154
expected = pd .array ([True , False , True , None ], dtype = "boolean" )
151
155
tm .assert_extension_array_equal (result , expected )
152
156
153
- result = pd .array (np .array ([1.0 , 0.0 , 1.0 , np .nan ]), dtype = "boolean" )
157
+
158
+ def test_to_boolean_array_integer_like ():
159
+ # integers of 0's and 1's
160
+ result = pd .array ([1 , 0 , 1 , 0 ], dtype = "boolean" )
161
+ expected = pd .array ([True , False , True , False ], dtype = "boolean" )
162
+ tm .assert_extension_array_equal (result , expected )
163
+
164
+ # with missing values
165
+ result = pd .array ([1 , 0 , 1 , None ], dtype = "boolean" )
166
+ expected = pd .array ([True , False , True , None ], dtype = "boolean" )
154
167
tm .assert_extension_array_equal (result , expected )
155
168
156
169
0 commit comments