json->dict

Author: dlawin

data_diff/__main__.py

@@ -411,7 +411,7 @@ def _main(
         # required to create this variable before get_stats
         diff_list = list(diff_iter)
         if json_output:
-            rich.print(json.dumps(diff_iter.get_stats_json()))
+            rich.print(json.dumps(diff_iter.get_stats_dict()))
         else:
             rich.print(diff_iter.get_stats_string())
 

data_diff/diff_tables.py

@@ -144,7 +144,7 @@ def get_stats_string(self):
 
         return string_output
 
-    def get_stats_json(self):
+    def get_stats_dict(self):
 
         diff_stats = self._get_stats()
         json_output = {

tests/test_api.py

@@ -91,13 +91,13 @@ def test_api_get_stats_string(self):
         t1.database.close()
         t2.database.close()
 
-    def test_api_get_stats_json(self):
+    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}}
         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)
         diff_list = list(diff)
-        output = diff.get_stats_json()
+        output = diff.get_stats_dict()
 
         self.assertEqual(expected_dict, output)
         self.assertIsNotNone(diff)