diff --git a/querybuilder/tests/helpers/test_schema_helpers.py b/querybuilder/tests/helpers/test_schema_helpers.py index f775dc3dc166d14dae7c01307ae59fa9fd41d8d1..a98c53f0b1c2dd8d837be4c62c560bcf361fd171 100644 --- a/querybuilder/tests/helpers/test_schema_helpers.py +++ b/querybuilder/tests/helpers/test_schema_helpers.py @@ -23,7 +23,7 @@ class TestColumnSpec(TestHelper): int, name="bar", relation_name="foo", schema_name="public", primary_key=True ) col = c.to_table_column() - assert type(col) == qb.atoms.columns.TableColumn + assert type(col) is qb.atoms.columns.TableColumn assert col.name == "bar" assert col.relation_name == "foo" assert col.schema_name == "public" @@ -41,7 +41,7 @@ class TestTableHelper(TestHelper): lastname: str birthday: date - assert type(mytable) == qb.atoms.relations.Table + assert type(mytable) is qb.atoms.relations.Table assert mytable.name == "mytable" assert mytable.schema_name is None assert len(mytable.columns) == 4 @@ -68,7 +68,7 @@ class TestTableHelper(TestHelper): lastname: str birthday: date - assert type(mytable) == qb.atoms.relations.Table + assert type(mytable) is qb.atoms.relations.Table assert mytable.name == "mytable" assert mytable.schema_name == "main" assert len(mytable.columns) == 4 @@ -94,7 +94,7 @@ class TestTableHelper(TestHelper): lastname: str birthday: date - assert type(mytable) == qb.atoms.relations.Table + assert type(mytable) is qb.atoms.relations.Table assert mytable.name == "break" assert mytable.schema_name is None assert len(mytable.columns) == 4 @@ -120,7 +120,7 @@ class TestTableHelper(TestHelper): lastname: str birthday: date - assert type(mytable) == qb.atoms.relations.Table + assert type(mytable) is qb.atoms.relations.Table assert mytable.name == "break" assert mytable.schema_name == "main" assert len(mytable.columns) == 4 @@ -253,7 +253,7 @@ class TestSchemaHelper(TestHelper): player: int since: date - assert type(main) == qb.atoms.schemas.Schema + assert type(main) is qb.atoms.schemas.Schema assert len(main.objects) == 3 self._check(main, "main") @@ -379,7 +379,7 @@ class TestSchemaHelper(TestHelper): # TODO: test index - assert type(main) == qb.atoms.schemas.Schema + assert type(main) is qb.atoms.schemas.Schema assert len(main.objects) == 5 assert len(main.relations) == 5 self._check(main, "main")