feat: fix unit tests

Author: pik94

data_diff/databases/mssql.py

@@ -38,7 +38,7 @@ def import_mssql():
 class Dialect(BaseDialect):
  name = "MsSQL"
  ROUNDS_ON_PREC_LOSS = True
- SUPPORTS_PRIMARY_KEY = True
+ SUPPORTS_PRIMARY_KEY: ClassVar[bool] = True
  SUPPORTS_INDEXES = True
  TYPE_CLASSES = {
  # Timestamps

data_diff/databases/mysql.py

@@ -40,7 +40,7 @@ def import_mysql():
 class Dialect(BaseDialect):
  name = "MySQL"
  ROUNDS_ON_PREC_LOSS = True
- SUPPORTS_PRIMARY_KEY = True
+ SUPPORTS_PRIMARY_KEY: ClassVar[bool] = True
  SUPPORTS_INDEXES = True
  TYPE_CLASSES = {
  # Dates

data_diff/databases/oracle.py

@@ -43,7 +43,7 @@ class Dialect(
  BaseDialect,
 ):
  name = "Oracle"
- SUPPORTS_PRIMARY_KEY = True
+ SUPPORTS_PRIMARY_KEY: ClassVar[bool] = True
  SUPPORTS_INDEXES = True
  TYPE_CLASSES: Dict[str, type] = {
  "NUMBER": Decimal,

data_diff/databases/postgresql.py

@@ -42,7 +42,7 @@ def import_postgresql():
 class PostgresqlDialect(BaseDialect):
  name = "PostgreSQL"
  ROUNDS_ON_PREC_LOSS = True
- SUPPORTS_PRIMARY_KEY = True
+ SUPPORTS_PRIMARY_KEY: ClassVar[bool] = True
  SUPPORTS_INDEXES = True
 
  TYPE_CLASSES: ClassVar[Dict[str, Type[ColType]]] = {

tests/test_joindiff.py

@@ -270,7 +270,9 @@ def test_null_pks(self):
  self.assertRaises(ValueError, list, x)
 
 
-@test_each_database_in_list(d for d in TEST_DATABASES if d.dialect.SUPPORTS_PRIMARY_KEY and d.SUPPORTS_UNIQUE_CONSTAINT)
+@test_each_database_in_list(
+ d for d in TEST_DATABASES if d.DIALECT_CLASS.SUPPORTS_PRIMARY_KEY and d.SUPPORTS_UNIQUE_CONSTAINT
+)
 class TestUniqueConstraint(DiffTestCase):
  def setUp(self):
  super().setUp()