TST: improve test_repr_bytes_61_lines setup to speed up test (#55724)

pandas-dev · Oct 27, 2023 · c36e302 · c36e302
1 parent b7ab856
commit c36e302
Showing 1 changed file with 1 addition and 6 deletions.
diff --git a/pandas/tests/frame/test_repr_info.py b/pandas/tests/frame/test_repr_info.py
@@ -29,12 +29,7 @@ class TestDataFrameReprInfoEtc:
     def test_repr_bytes_61_lines(self):
         # GH#12857
         lets = list("ACDEFGHIJKLMNOP")
-        slen = 50
-        nseqs = 1000
-        words = [
-            [np.random.default_rng(2).choice(lets) for x in range(slen)]
-            for _ in range(nseqs)
-        ]
+        words = np.random.default_rng(2).choice(lets, (1000, 50))
         df = DataFrame(words).astype("U1")
         assert (df.dtypes == object).all()