use List[int] from typing

canonical · Jan 26, 2024 · 840cadc · 840cadc
1 parent ee5a19b
commit 840cadc
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/unit/test_collect_schemas.py b/tests/unit/test_collect_schemas.py
@@ -97,7 +97,10 @@ class ProviderSchema(DataBagSchema):
     (
         (dedent("""Foo2: int=1"""), "Foo2"),
         (dedent("""Bar: str='baz'"""), "Bar"),
-        (dedent("""Baz: list=[1,2,3]"""), "Baz"),
+        (dedent("""
+        from typing import List
+        
+        Baz: List[int]=[1,2,3]"""), "Baz"),
     ),
 )
 def test_get_schema_from_module_wrong_type(tmp_path, schema_source, schema_name):