Merge pull request #497 from dlawin/tabulate_for_rows_added_removed

cleaner rows added/removed

Author: dlawin

data_diff/utils.py

@@ -5,6 +5,7 @@
 import operator
 import threading
 from datetime import datetime
+from tabulate import tabulate
 
 
 def safezip(*args):
@@ -132,12 +133,9 @@ def __repr__(self) -> str:
 def dbt_diff_string_template(
     rows_added: str, rows_removed: str, rows_updated: str, rows_unchanged: str, extra_info_dict: Dict, extra_info_str
 ) -> str:
-    string_output = "\n| Rows Added\t| Rows Removed\n"
-    string_output += "------------------------------------------------------------\n"
+    string_output = f"\n{tabulate([[rows_added, rows_removed]], headers=['Rows Added', 'Rows Removed'])}"
 
-    string_output += f"| {rows_added}\t\t| {rows_removed}\n"
-    string_output += "------------------------------------------------------------\n\n"
-    string_output += f"Updated Rows: {rows_updated}\n"
+    string_output += f"\n\nUpdated Rows: {rows_updated}\n"
     string_output += f"Unchanged Rows: {rows_unchanged}\n\n"
 
     string_output += extra_info_str

poetry.lock

@@ -44,6 +44,7 @@ duckdb = {version="^0.7.0", optional=true}
 dbt-artifacts-parser = {version="^0.2.5", optional=true}
 dbt-core = {version="^1.0.0", optional=true}
 keyring = "*"
+tabulate = "^0.9.0"
 
 [tool.poetry.dev-dependencies]
 parameterized = "*"

pyproject.toml

@@ -420,7 +420,7 @@ def test_integration_basic_dbt(self):
             # 4 with no diffs
             assert diff_string.count("No row differences") == 4
             # 1 with a diff
-            assert diff_string.count("| Rows Added    | Rows Removed") == 1
+            assert diff_string.count("  Rows Added    Rows Removed") == 1
 
     def test_integration_cloud_dbt(self):
         project_dir = os.environ.get("DATA_DIFF_DBT_PROJ")