File tree 1 file changed +44
-0
lines changed
1 file changed +44
-0
lines changed Original file line number Diff line number Diff line change
1
+ import pandas as pd
2
+
3
+
4
+ class MyDtype (pd .api .extensions .ExtensionDtype ):
5
+ name = "mydtype"
6
+ type = list
7
+
8
+
9
+ class MyEA (pd .api .extensions .ExtensionArray ):
10
+ def __init__ (self , data ):
11
+ self .data = data
12
+ self ._dtype = MyDtype ()
13
+
14
+ @property
15
+ def dtype (self ):
16
+ return self ._dtype
17
+
18
+ def __len__ (self ):
19
+ return 1
20
+
21
+ def __array__ (self , dtype = None ):
22
+ raise ValueError ("Cannot be converted to an array!" )
23
+
24
+ def _format_array (
25
+ self ,
26
+ formatter : None ,
27
+ float_format : None ,
28
+ na_rep = "NaN" ,
29
+ digits = None ,
30
+ space = None ,
31
+ justify = "right" ,
32
+ decimal = "." ,
33
+ leading_space = True ,
34
+ quoting = None ,
35
+ ):
36
+ return ["<MyEA>([1])" ]
37
+
38
+
39
+ def test_no_conversion ():
40
+ s = pd .Series (MyEA ([1 ]))
41
+ repr (s ) # OK!
42
+
43
+ df = pd .DataFrame ({"A" : MyEA ([1 ])}, copy = False )
44
+ repr (df ) # OK!
You can’t perform that action at this time.
0 commit comments