Skip to content

Commit f551ecb

Browse files
author
Tom Augspurger
committed
added specific test
1 parent 3885851 commit f551ecb

File tree

1 file changed

+44
-0
lines changed

1 file changed

+44
-0
lines changed

pandas/tests/extension/test_format.py

+44
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,44 @@
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!

0 commit comments

Comments
 (0)