summaryrefslogtreecommitdiff
path: root/tests/test_autogen_composition.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/test_autogen_composition.py')
-rw-r--r--tests/test_autogen_composition.py10
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",
+ },
)