From 840cadc172c415c0d6f9be15e7abd0972bceb2ce Mon Sep 17 00:00:00 2001
From: Mateusz Kulewicz <mateusz.kulewicz@canonical.com>
Date: Fri, 26 Jan 2024 12:07:33 +0100
Subject: [PATCH] use List[int] from typing

---
 tests/unit/test_collect_schemas.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/tests/unit/test_collect_schemas.py b/tests/unit/test_collect_schemas.py
index e87d57b..d7e9138 100644
--- 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):