Import 'Literal' from serde.compat

yukinarit · Jul 24, 2022 · a7450c1 · a7450c1
1 parent eddf250
commit a7450c1
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 7 deletions.
diff --git a/tests/test_compat.py b/tests/test_compat.py
@@ -3,14 +3,9 @@
 from datetime import datetime
 from typing import Dict, Generic, List, NewType, Optional, Set, Tuple, TypeVar, Union
 
-if sys.version_info[:2] == (3, 7):
-    from typing_extensions import Literal
-else:
-    from typing import Literal
-
-
 import serde
 from serde.compat import (
+    Literal,
     is_dict,
     is_generic,
     is_list,

diff --git a/tests/test_union.py b/tests/test_union.py
@@ -2,14 +2,15 @@
 import sys
 from dataclasses import dataclass
 from ipaddress import IPv4Address
-from typing import Dict, Generic, List, Literal, Optional, Tuple, TypeVar, Union
+from typing import Dict, Generic, List, Optional, Tuple, TypeVar, Union
 from uuid import UUID
 
 import pytest
 
 from serde import SerdeError, from_dict, from_tuple
 from serde import init as serde_init
 from serde import logger, serde, to_dict, to_tuple
+from serde.compat import Literal
 from serde.json import from_json, to_json
 
 logging.basicConfig(level=logging.WARNING)