TST: tests for needs-test issues #12857, #12689

pandas-dev · jreback · Dec 24, 2019 · Dec 16, 2019 · Dec 16, 2019 · Dec 18, 2019
commit 74fc0cc39fcaef1c162006268d1416d38190e260
diff --git a/pandas/tests/frame/test_arithmetic.py b/pandas/tests/frame/test_arithmetic.py
@@ -16,6 +16,13 @@
 class TestFrameComparisons:
     # Specifically _not_ flex-comparisons
 
+    def test_frame_in_list(self):
+        # GH#12689 this should raise at the DataFrame level, not blocks
+        df = pd.DataFrame(np.random.randn(6,4), columns=list('ABCD'))
+        msg = "The truth value of a DataFrame is ambiguous"
+        with pytest.raises(ValueError, match=msg):
+            df in [None]
+
     def test_comparison_invalid(self):
         def check(df, df2):
 

diff --git a/pandas/tests/frame/test_repr_info.py b/pandas/tests/frame/test_repr_info.py
@@ -1,5 +1,6 @@
 from datetime import datetime, timedelta
 from io import StringIO
+import random
 import re
 import sys
 import textwrap
@@ -27,6 +28,17 @@
 
 
 class TestDataFrameReprInfoEtc:
+
+    def test_repr_bytes_61_lines(self):
+        # GH#12857
+        lets = 'ACDEFGHIJKLMNOP'
+        slen = 50
+        nseqs = 1000
+        words = [[random.choice(lets) for x in range(slen)] for _ in range(nseqs)]
+        df = pd.DataFrame(words).astype('S1')
+        assert (df.dtypes == 'S1').all()
+        repr(df)  # smoke test
+
     def test_repr_empty(self):
         # empty
         foo = repr(DataFrame())  # noqa