@@ -9,6 +9,11 @@ def setup(self):
99 self .values_float = np .array ([1.0 , 0.0 , 1.0 , 0.0 ])
1010 self .values_integer = np .array ([1 , 0 , 1 , 0 ])
1111 self .values_integer_like = [1 , 0 , 1 , 0 ]
12+ self .data = np .array ([True , False , True , False ])
13+ self .mask = np .array ([False , False , True , False ])
14+
15+ def time_constructor (self ):
16+ pd .arrays .BooleanArray (self .data , self .mask )
1217
1318 def time_from_bool_array (self ):
1419 pd .array (self .values_bool , dtype = "boolean" )
@@ -21,3 +26,16 @@ def time_from_integer_like(self):
2126
2227 def time_from_float_array (self ):
2328 pd .array (self .values_float , dtype = "boolean" )
29+
30+
31+ class IntegerArray :
32+ def setup (self ):
33+ self .values_integer = np .array ([1 , 0 , 1 , 0 ])
34+ self .data = np .array ([1 , 2 , 3 , 4 ], dtype = "int64" )
35+ self .mask = np .array ([False , False , True , False ])
36+
37+ def time_constructor (self ):
38+ pd .arrays .IntegerArray (self .data , self .mask )
39+
40+ def time_from_integer_array (self ):
41+ pd .array (self .values_integer , dtype = "Int64" )
0 commit comments