Move all literals test to test_utils

pydantic · May 23, 2020 · 6949f5b · 6949f5b
1 parent e339856
commit 6949f5b
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 18 deletions.
diff --git a/tests/test_typing.py b/tests/test_typing.py
diff --git a/tests/test_utils.py b/tests/test_utils.py
@@ -11,7 +11,7 @@
 from pydantic.color import Color
 from pydantic.dataclasses import dataclass
 from pydantic.fields import Undefined
-from pydantic.typing import display_as_type, is_new_type, new_type_supertype
+from pydantic.typing import Literal, all_literal_values, display_as_type, is_new_type, new_type_supertype
 from pydantic.utils import (
     ClassAttribute,
     ValueItems,
@@ -320,3 +320,16 @@ class Foo:
     f = Foo()
     f.attr = 'not foo'
     assert f.attr == 'not foo'
+
+
+@pytest.mark.skipif(not Literal, reason='typing_extensions not installed')
+def test_all_literal_values():
+    L1 = Literal['1']
+    assert all_literal_values(L1) == ('1',)
+
+    L2 = Literal['2']
+    L12 = Literal[L1, L2]
+    assert sorted(all_literal_values(L12)) == sorted(('1', '2'))
+
+    L312 = Literal['3', Literal[L1, L2]]
+    assert sorted(all_literal_values(L312)) == sorted(('1', '2', '3'))