diff options
Diffstat (limited to 'tests/test_external_dialect.py')
-rw-r--r-- | tests/test_external_dialect.py | 32 |
1 files changed, 13 insertions, 19 deletions
diff --git a/tests/test_external_dialect.py b/tests/test_external_dialect.py index 9ddc12f..de66517 100644 --- a/tests/test_external_dialect.py +++ b/tests/test_external_dialect.py @@ -65,7 +65,7 @@ class EXT_ARRAY(sqla_types.TypeEngine): if isinstance(item_type, type): item_type = item_type() self.item_type = item_type - super(EXT_ARRAY, self).__init__() + super().__init__() class FOOBARTYPE(sqla_types.TypeEngine): @@ -94,12 +94,10 @@ class ExternalDialectRenderTest(TestBase): eq_( self.autogen_context.imports, - set( - [ - "from tests.test_external_dialect " - "import custom_dialect_types" - ] - ), + { + "from tests.test_external_dialect " + "import custom_dialect_types" + }, ) def test_external_nested_render_sqla_type(self): @@ -121,12 +119,10 @@ class ExternalDialectRenderTest(TestBase): eq_( self.autogen_context.imports, - set( - [ - "from tests.test_external_dialect " - "import custom_dialect_types" - ] - ), + { + "from tests.test_external_dialect " + "import custom_dialect_types" + }, ) def test_external_nested_render_external_type(self): @@ -141,10 +137,8 @@ class ExternalDialectRenderTest(TestBase): eq_( self.autogen_context.imports, - set( - [ - "from tests.test_external_dialect " - "import custom_dialect_types" - ] - ), + { + "from tests.test_external_dialect " + "import custom_dialect_types" + }, ) |