lint

Author: dlawin

data_diff/diff_tables.py

@@ -78,6 +78,7 @@ def _run_in_background(self, *funcs):
             for f in futures:
                 f.result()
 
+
 @dataclass
 class DiffStats:
     diff_by_sign: dict[str, int]
@@ -86,6 +87,7 @@ class DiffStats:
     unchanged: int
     diff_percent: float
 
+
 @dataclass
 class DiffResultWrapper:
     diff: iter  # DiffResult
@@ -145,18 +147,19 @@ def get_stats_dict(self):
 
         diff_stats = self._get_stats()
         json_output = {
-                    "rows_A": diff_stats.table1_count,
-                    "rows_B": diff_stats.table2_count,
-                    "exclusive_A": diff_stats.diff_by_sign["-"],
-                    "exclusive_B": diff_stats.diff_by_sign["+"],
-                    "updated": diff_stats.diff_by_sign["!"],
-                    "unchanged": diff_stats.unchanged,
-                    "total": sum(diff_stats.diff_by_sign.values()),
-                    "stats": self.stats,
-                }
+            "rows_A": diff_stats.table1_count,
+            "rows_B": diff_stats.table2_count,
+            "exclusive_A": diff_stats.diff_by_sign["-"],
+            "exclusive_B": diff_stats.diff_by_sign["+"],
+            "updated": diff_stats.diff_by_sign["!"],
+            "unchanged": diff_stats.unchanged,
+            "total": sum(diff_stats.diff_by_sign.values()),
+            "stats": self.stats,
+        }
 
         return json_output
 
+
 class TableDiffer(ThreadBase, ABC):
     bisection_factor = 32
     stats: dict = {}

tests/test_api.py

@@ -89,9 +89,18 @@ def test_api_get_stats_string(self):
 
         t1.database.close()
         t2.database.close()
-    
+
     def test_api_get_stats_dict(self):
-        expected_dict = {'rows_A': 5, 'rows_B': 4, 'exclusive_A': 1, 'exclusive_B': 0, 'updated': 0, 'unchanged': 4, 'total': 1, 'stats': {'rows_downloaded': 5}}
+        expected_dict = {
+            "rows_A": 5,
+            "rows_B": 4,
+            "exclusive_A": 1,
+            "exclusive_B": 0,
+            "updated": 0,
+            "unchanged": 4,
+            "total": 1,
+            "stats": {"rows_downloaded": 5},
+        }
         t1 = connect_to_table(TEST_MYSQL_CONN_STRING, self.table_src_name)
         t2 = connect_to_table(TEST_MYSQL_CONN_STRING, self.table_dst_name)
         diff = diff_tables(t1, t2)

tests/test_cli.py

@@ -68,7 +68,9 @@ def tearDown(self) -> None:
         return super().tearDown()
 
     def test_basic(self):
-        diff = run_datadiff_cli(TEST_MYSQL_CONN_STRING, self.table_src_name, TEST_MYSQL_CONN_STRING, self.table_dst_name)
+        diff = run_datadiff_cli(
+            TEST_MYSQL_CONN_STRING, self.table_src_name, TEST_MYSQL_CONN_STRING, self.table_dst_name
+        )
         assert len(diff) == 1
 
     def test_options(self):
@@ -95,13 +97,13 @@ def test_stats(self):
             TEST_MYSQL_CONN_STRING, self.table_src_name, TEST_MYSQL_CONN_STRING, self.table_dst_name, "-s"
         )
         assert len(diff_output) == 11
-    
+
     def test_stats_json(self):
         diff_output = run_datadiff_cli(
             TEST_MYSQL_CONN_STRING, self.table_src_name, TEST_MYSQL_CONN_STRING, self.table_dst_name, "-s", "--json"
         )
         assert len(diff_output) == 2
-    
+
     def test_stats_no_diff(self):
         diff_output = run_datadiff_cli(
             TEST_MYSQL_CONN_STRING, self.table_src_name, TEST_MYSQL_CONN_STRING, self.table_src_name, "-s"