Ran black

Author: erezsh

data_diff/databases/clickhouse.py

@@ -69,7 +69,7 @@ def cursor(self, cursor_factory=None):
     def _parse_type_repr(self, type_repr: str) -> Optional[Type[ColType]]:
         nullable_prefix = "Nullable("
         if type_repr.startswith(nullable_prefix):
-            type_repr = type_repr[len(nullable_prefix):].rstrip(")")
+            type_repr = type_repr[len(nullable_prefix) :].rstrip(")")
 
         if type_repr.startswith("Decimal"):
             type_repr = "Decimal"
@@ -91,7 +91,7 @@ def to_string(self, s: str) -> str:
         return f"toString({s})"
 
     def normalize_timestamp(self, value: str, coltype: TemporalType) -> str:
-        prec= coltype.precision
+        prec = coltype.precision
         if coltype.rounds:
             timestamp = f"toDateTime64(round(toUnixTimestamp64Micro(toDateTime64({value}, 6)) / 1000000, {prec}), 6)"
             return self.to_string(timestamp)

tests/common.py

@@ -21,7 +21,7 @@
 TEST_ORACLE_CONN_STRING: str = None
 TEST_DATABRICKS_CONN_STRING: str = os.environ.get("DATADIFF_DATABRICKS_URI")
 TEST_TRINO_CONN_STRING: str = os.environ.get("DATADIFF_TRINO_URI") or None
- # clickhouse uri for provided docker - "clickhouse://clickhouse:Password1@localhost:9000/clickhouse"
+# clickhouse uri for provided docker - "clickhouse://clickhouse:Password1@localhost:9000/clickhouse"
 TEST_CLICKHOUSE_CONN_STRING: str = os.environ.get("DATADIFF_CLICKHOUSE_URI") or None
 
 DEFAULT_N_SAMPLES = 50

tests/test_database_types.py

@@ -271,8 +271,8 @@ def init_conns():
         ],
         "uuid": [
             "String",
-        ]
-    }
+        ],
+    },
 }
 
 
@@ -482,13 +482,13 @@ def _insert_to_table(conn, table, values, type):
             if type.startswith("DateTime64"):
                 value = f"'{sample.replace(tzinfo=None)}'"
 
-            elif type == 'DateTime':
+            elif type == "DateTime":
                 sample = sample.replace(tzinfo=None)
                 # Clickhouse's DateTime does not allow to store micro/milli/nano seconds
                 value = f"'{str(sample)[:19]}'"
 
-            elif type.startswith('Decimal'):
-                precision = int(type[8:].rstrip(')').split(',')[1])
+            elif type.startswith("Decimal"):
+                precision = int(type[8:].rstrip(")").split(",")[1])
                 value = round(sample, precision)
 
             else:

tests/test_diff_tables.py

@@ -386,7 +386,6 @@ def test_diff_sorted_by_key(self):
 
 @test_per_database
 class TestDiffTables2(TestPerDatabase):
-
     def test_diff_column_names(self):
         float_type = _get_float_type(self.connection)