diff --git a/data_diff/diff_tables.py b/data_diff/diff_tables.py index aaf56c9c..c60ac186 100644 --- a/data_diff/diff_tables.py +++ b/data_diff/diff_tables.py @@ -3,6 +3,7 @@ import re import time +import humanfriendly from abc import ABC, abstractmethod from enum import Enum from contextlib import contextmanager @@ -135,15 +136,22 @@ def _get_stats(self, is_dbt: bool = False) -> DiffStats: return DiffStats(diff_by_sign, table1_count, table2_count, unchanged, diff_percent, extra_column_diffs) - def get_stats_string(self, is_dbt: bool = False): - diff_stats = self._get_stats(is_dbt) + def __init__(self, diff_by_sign, table1_count, table2_count, unchanged, diff_percent, extra_column_diffs=None): + self.diff_by_sign = diff_by_sign + self.table1_count = table1_count + self.table2_count = table2_count + self.unchanged = unchanged + self.diff_percent = diff_percent + self.extra_column_diffs = extra_column_diffs + def get_stats_string(self, is_dbt=False): + diff_stats = self._get_stats(is_dbt) if is_dbt: string_output = dbt_diff_string_template( - diff_stats.diff_by_sign["-"], - diff_stats.diff_by_sign["+"], - diff_stats.diff_by_sign["!"], - diff_stats.unchanged, + humanfriendly.format_number(diff_stats.diff_by_sign["-"]), + humanfriendly.format_number(diff_stats.diff_by_sign["+"]), + humanfriendly.format_number(diff_stats.diff_by_sign["!"]), + humanfriendly.format_number(diff_stats.unchanged), diff_stats.extra_column_diffs, "Values Updated:", ) diff --git a/docs/requirements.txt b/docs/requirements.txt index 252c7acb..3144a7a2 100644 --- a/docs/requirements.txt +++ b/docs/requirements.txt @@ -1,4 +1,5 @@ # https://docs.readthedocs.io/en/stable/guides/specifying-dependencies.html#specifying-a-requirements-file +humanfriendly sphinx-gallery sphinx_markdown_tables sphinx-copybutton