Fix test for py>3.7

bogdandm · Mar 10, 2022 · fe0441e · fe0441e
1 parent 6674f7f
commit fe0441e
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/test/test_code_generation/test_models_code_generator.py b/test/test_code_generation/test_models_code_generator.py
@@ -171,7 +171,7 @@ class Test:
             "f": StringLiteral({'with \\ // slash'}),
         }),
         "fields": {
-            "imports": "from typing_extensions import Literal",
+            "imports": f"{LITERAL_SOURCE} import Literal",
             "fields": [
                 'a: Literal["basic"]',
                 'b: Literal["with space"]',
@@ -181,8 +181,8 @@ class Test:
                 'f: Literal["with \\\\ // slash"]'
             ]
         },
-        "generated": trim("""
-        from typing_extensions import Literal
+        "generated": trim(f"""
+        {LITERAL_SOURCE} import Literal
 
 
         class Test: