From fe0441e46d4e44fc482f1a71fed351404758cfb2 Mon Sep 17 00:00:00 2001
From: bogdan_dm <bogdan.dm1995@yandex.ru>
Date: Thu, 10 Mar 2022 12:20:58 +0300
Subject: [PATCH] Fix test for py>3.7

---
 test/test_code_generation/test_models_code_generator.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/test/test_code_generation/test_models_code_generator.py b/test/test_code_generation/test_models_code_generator.py
index b3758de..a29547f 100644
--- 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: