showcase a use case

Signed-off-by: Jan Kowalleck <jan.kowalleck@gmail.com>
madpah · Oct 5, 2023 · 3ae2d51 · 3ae2d51
1 parent cf905b3
commit 3ae2d51
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/tests/model.py b/tests/model.py
@@ -20,7 +20,7 @@
 import re
 from datetime import date
 from enum import Enum, unique
-from typing import Iterable, List, Optional, Set
+from typing import Dict, Iterable, List, Optional, Set, Type
 from uuid import UUID, uuid4
 
 import serializable
@@ -48,6 +48,14 @@ class SchemaVersion4(ViewType):
     pass
 
 
+SCHEMAVERSION_MAP: Dict[int, Type[ViewType]] = {
+    1: SchemaVersion1,
+    2: SchemaVersion2,
+    3: SchemaVersion3,
+    4: SchemaVersion4,
+}
+
+
 class ReferenceReferences(BaseHelper):
 
     @classmethod