diff options
Diffstat (limited to 'tests/test_autogen_composition.py')
-rw-r--r-- | tests/test_autogen_composition.py | 10 |
1 files changed, 4 insertions, 6 deletions
diff --git a/tests/test_autogen_composition.py b/tests/test_autogen_composition.py index acd3603..99e5486 100644 --- a/tests/test_autogen_composition.py +++ b/tests/test_autogen_composition.py @@ -243,12 +243,10 @@ nullable=True)) autogenerate._render_migration_diffs(self.context, template_args) eq_( set(template_args["imports"].split("\n")), - set( - [ - "from foobar import bat", - "from mypackage import my_special_import", - ] - ), + { + "from foobar import bat", + "from mypackage import my_special_import", + }, ) |