fix test type test

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")