Fix Literal import in tests

Author: bogdandm

test/test_code_generation/test_models_code_generator.py

@@ -1,12 +1,15 @@
 from typing import Dict, List, Type, Union
 
 import pytest
+from typing_extensions import Literal
 
 from json_to_models.dynamic_typing import (AbsoluteModelRef, BaseType, DDict, DList, DOptional, IntString, IsoDateString, ModelMeta, ModelPtr, StringLiteral, StringSerializable, Unknown, compile_imports)
 from json_to_models.models.base import GenericModelCodeGenerator, generate_code
 from json_to_models.models.structure import sort_fields
 from json_to_models.models.utils import indent
 
+LITERAL_SOURCE = f"from {Literal.__module__}"
+
 # Data structure:
 # (string, indent lvl, indent string)
 # result
@@ -305,8 +308,8 @@ def test_unicode(value: ModelMeta, kwargs: dict, expected: str):
             "bar": StringLiteral({'bar', 'foo'})
         }),
         {},
-        trim("""
-        from typing_extensions import Literal
+        trim(f"""
+        {LITERAL_SOURCE} import Literal
         
         
         class M: