Add a test case where the number of rows exceeds the limit of 10

Signed-off-by: Baden Ashford <[email protected]>
unionai-oss · Jan 25, 2024 · 1bb9b0c · 1bb9b0c
1 parent 1ffbeca
commit 1bb9b0c
Show file tree

Hide file tree

Showing 2 changed files with 40 additions and 2 deletions.
diff --git a/pandera/backends/pandas/error_formatters.py b/pandera/backends/pandas/error_formatters.py
@@ -345,9 +345,18 @@ def dataframe_to_tabular_formatted_string(
     # Bottom border and additional rows message if applicable
     if len(dataframe) > limit:
         formatted_string += (
-            "│ " + " ┆ ".join(["…" for _ in column_widths.values()]) + " │\n"
+            "│"
+            + "│".join(
+                [" " + "…" * (width) + " " for width in column_widths.values()]
+            )
+            + "│\n"
+        )
+        formatted_string += (
+            "└"
+            + "┴".join(["─" * (width + 2) for width in column_widths.values()])
+            + "┘\n"
         )
-        formatted_string += f"And {len(dataframe) - limit} others"
+        formatted_string += f"And {len(dataframe) - limit} more rows\n"
     else:
         formatted_string += (
             "└"

diff --git a/tests/core/test_error_formatters.py b/tests/core/test_error_formatters.py
@@ -87,6 +87,35 @@ def _mock_custom_wide_check(df):
 │ None  │ Wide Schema │ _mock_custom_wide_ch │ False        │ DataFrameSchema │ 0            │
 │       │             │ eck                  │              │                 │              │
 └───────┴─────────────┴──────────────────────┴──────────────┴─────────────────┴──────────────┘
+""",
+        ),
+        (
+            DataFrameSchema(
+                {"count": Column(int, checks=[Check.greater_than(1)])}
+            ),
+            pd.DataFrame(
+                {
+                    "count": [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
+                }
+            ),
+            """
+Schema Nameless Schema: A total of 16 schema errors were found.
+┌───────┬────────┬─────────────────┬──────────────┬────────────────┬──────────────┐
+│ index ┆ column ┆      check      ┆ failure_case ┆ schema_context ┆ check_number │
+╞═══════╪════════╪═════════════════╪══════════════╪════════════════╪══════════════╡
+│ 0     │ count  │ greater_than(1) │ 0            │ Column         │ 0            │
+│ 1     │ count  │ greater_than(1) │ 0            │ Column         │ 0            │
+│ 2     │ count  │ greater_than(1) │ 0            │ Column         │ 0            │
+│ 3     │ count  │ greater_than(1) │ 0            │ Column         │ 0            │
+│ 4     │ count  │ greater_than(1) │ 0            │ Column         │ 0            │
+│ 5     │ count  │ greater_than(1) │ 0            │ Column         │ 0            │
+│ 6     │ count  │ greater_than(1) │ 0            │ Column         │ 0            │
+│ 7     │ count  │ greater_than(1) │ 0            │ Column         │ 0            │
+│ 8     │ count  │ greater_than(1) │ 0            │ Column         │ 0            │
+│ 9     │ count  │ greater_than(1) │ 0            │ Column         │ 0            │
+│ …………… │ ……………… │ ……………………………………… │ ……………………………… │ …………………………………… │ ……………………………… │
+└───────┴────────┴─────────────────┴──────────────┴────────────────┴──────────────┘
+And 6 more rows
 """,
         ),
     ],