@@ -9,6 +9,11 @@ def setup(self):
9
9
self .values_float = np .array ([1.0 , 0.0 , 1.0 , 0.0 ])
10
10
self .values_integer = np .array ([1 , 0 , 1 , 0 ])
11
11
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 )
12
17
13
18
def time_from_bool_array (self ):
14
19
pd .array (self .values_bool , dtype = "boolean" )
@@ -21,3 +26,16 @@ def time_from_integer_like(self):
21
26
22
27
def time_from_float_array (self ):
23
28
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