black

Author: erezsh

data_diff/databases/base.py

@@ -233,7 +233,6 @@ def query_table_schema(self, path: DbPath) -> Dict[str, tuple]:
         assert len(d) == len(rows)
         return d
 
-
     def select_table_unique_columns(self, path: DbPath) -> str:
         schema, table = self._normalize_table_path(path)
 

data_diff/databases/database_types.py

@@ -260,8 +260,7 @@ def select_table_unique_columns(self, path: DbPath) -> str:
 
     @abstractmethod
     def query_table_unique_columns(self, path: DbPath) -> List[str]:
-        """Query the table for its unique columns for table in 'path', and return {column}
-        """
+        """Query the table for its unique columns for table in 'path', and return {column}"""
         ...
 
     @abstractmethod

data_diff/joindiff_tables.py

@@ -203,7 +203,7 @@ def _test_duplicate_keys(self, table1: TableSegment, table2: TableSegment):
             unvalidated = list(set(key_columns) - set(unique))
             if unvalidated:
                 # Validate that there are no duplicate keys
-                self.stats['validated_unique_keys'] = self.stats.get('validated_unique_keys', []) + [unvalidated]
+                self.stats["validated_unique_keys"] = self.stats.get("validated_unique_keys", []) + [unvalidated]
                 q = t.select(total=Count(), total_distinct=Count(Concat(this[unvalidated]), distinct=True))
                 total, total_distinct = ts.database.query(q, tuple)
                 if total != total_distinct:

data_diff/queries/ast_classes.py

@@ -655,7 +655,11 @@ def compile(self, c: Compiler) -> str:
             return f"CREATE TABLE {ne}{c.compile(self.path)} AS {c.compile(self.source_table)}"
 
         schema = ", ".join(f"{c.database.quote(k)} {c.database.type_repr(v)}" for k, v in self.path.schema.items())
-        pks = ", PRIMARY KEY (%s)" % ', '.join(self.primary_keys) if self.primary_keys and c.database.SUPPORTS_PRIMARY_KEY else ""
+        pks = (
+            ", PRIMARY KEY (%s)" % ", ".join(self.primary_keys)
+            if self.primary_keys and c.database.SUPPORTS_PRIMARY_KEY
+            else ""
+        )
         return f"CREATE TABLE {ne}{c.compile(self.path)}({schema}{pks})"
 
 

data_diff/query_utils.py

@@ -8,6 +8,7 @@
 from .databases import Oracle
 from .queries import table, commit, Expr
 
+
 def _drop_table_oracle(name: DbPath):
     t = table(name)
     # Experience shows double drop is necessary
@@ -50,6 +51,7 @@ def _append_to_table(path: DbPath, expr: Expr):
     yield t.insert_expr(expr)
     yield commit
 
+
 def append_to_table(db, path, expr):
     f = _append_to_table_oracle if isinstance(db, Oracle) else _append_to_table
     db.query(f(path, expr))

tests/common.py

@@ -120,7 +120,6 @@ def str_to_checksum(str: str):
     return int(md5[half_pos:], 16)
 
 
-
 class TestPerDatabase(unittest.TestCase):
     db_cls = None
 

tests/test_diff_tables.py

@@ -383,9 +383,7 @@ def test_string_keys(self):
         diff = list(differ.diff_tables(self.a, self.b))
         self.assertEqual(diff, [("-", (str(self.new_uuid), "This one is different"))])
 
-        self.connection.query(
-            self.src_table.insert_row('unexpected', '<-- this bad value should not break us')
-        )
+        self.connection.query(self.src_table.insert_row("unexpected", "<-- this bad value should not break us"))
 
         self.assertRaises(ValueError, list, differ.diff_tables(self.a, self.b))
 
@@ -421,7 +419,7 @@ def setUp(self):
             src_table.create(),
             src_table.insert_rows(values),
             table(self.table_dst_path).create(src_table),
-            src_table.insert_row(self.new_alphanum, 'This one is different'),
+            src_table.insert_row(self.new_alphanum, "This one is different"),
             commit,
         ]
 
@@ -491,7 +489,7 @@ def test_varying_alphanum_keys(self):
         self.assertEqual(diff, [("-", (str(self.new_alphanum), "This one is different"))])
 
         self.connection.query(
-            self.src_table.insert_row('@@@', '<-- this bad value should not break us'),
+            self.src_table.insert_row("@@@", "<-- this bad value should not break us"),
             commit,
         )
 
@@ -548,13 +546,15 @@ def setUp(self):
 
         self.null_uuid = uuid.uuid1(32132131)
 
-        self.connection.query([
-            src_table.create(),
-            src_table.insert_rows(values),
-            table(self.table_dst_path).create(src_table),
-            src_table.insert_row(self.null_uuid, None),
-            commit,
-        ])
+        self.connection.query(
+            [
+                src_table.create(),
+                src_table.insert_rows(values),
+                table(self.table_dst_path).create(src_table),
+                src_table.insert_row(self.null_uuid, None),
+                commit,
+            ]
+        )
 
         self.a = _table_segment(self.connection, self.table_src_path, "id", "text_comment", case_sensitive=False)
         self.b = _table_segment(self.connection, self.table_dst_path, "id", "text_comment", case_sensitive=False)
@@ -576,9 +576,9 @@ def setUp(self):
         self.connection.query(
             [
                 src_table.create(),
-                src_table.insert_row(uuid.uuid1(1), '1'),
+                src_table.insert_row(uuid.uuid1(1), "1"),
                 table(self.table_dst_path).create(src_table),
-                src_table.insert_row(self.null_uuid, None),       # Add a row where a column has NULL value
+                src_table.insert_row(self.null_uuid, None),  # Add a row where a column has NULL value
                 commit,
             ]
         )

tests/test_joindiff.py

@@ -281,11 +281,8 @@ def setUp(self):
             schema={"id": int, "userid": int, "movieid": int, "rating": float},
         )
 
-        self.connection.query([
-            self.src_table.create(primary_keys=['id']),
-            self.dst_table.create(primary_keys=['id', 'userid']),
-            commit
-            ]
+        self.connection.query(
+            [self.src_table.create(primary_keys=["id"]), self.dst_table.create(primary_keys=["id", "userid"]), commit]
         )
 
         self.differ = JoinDiffer()
@@ -305,12 +302,12 @@ def test_unique_constraint(self):
 
         res = list(self.differ.diff_tables(table, table2))
         assert not res
-        assert 'validated_unique_keys' not in self.differ.stats
+        assert "validated_unique_keys" not in self.differ.stats
 
         # Test active validation
         table = TableSegment(self.connection, self.table_src_path, ("userid",), case_sensitive=False)
         table2 = TableSegment(self.connection, self.table_dst_path, ("userid",), case_sensitive=False)
 
         res = list(self.differ.diff_tables(table, table2))
         assert not res
-        self.assertEqual( self.differ.stats['validated_unique_keys'], [['userid']] )
+        self.assertEqual(self.differ.stats["validated_unique_keys"], [["userid"]])