add repr tests

pandas-dev · Jun 3, 2018 · 4c4262e · 4c4262e
1 parent 90400eb
commit 4c4262e
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/pandas/tests/extension/base/interface.py b/pandas/tests/extension/base/interface.py
@@ -40,6 +40,16 @@ def test_repr(self, data):
         df = pd.DataFrame({"A": data})
         repr(df)
 
+    def test_repr_array(self, data):
+        # some arrays may be able to assert
+        # attributes in the repr
+        repr(data)
+
+    def test_repr_array_long(self, data):
+        # some arrays may be able to assert a ... in the repr
+        with pd.option_context('display.max_seq_items', 1):
+            repr(data)
+
     def test_dtype_name_in_info(self, data):
         buf = StringIO()
         pd.DataFrame({"A": data}).info(buf=buf)