Add a test for the class refatt map to ensure model changes don't bre…

…ak the enref functions
ga4gh · Sep 18, 2023 · 37cbdfd · 37cbdfd
1 parent b8c6fd2
commit 37cbdfd
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/tests/test_vrs2.py b/tests/test_vrs2.py
@@ -199,3 +199,15 @@ def test_enref():
         'end': 128325835})
     assert dereffed.location.model_dump(exclude_none=True) == allele_383650.location.model_dump(exclude_none=True)
     assert dereffed.model_dump() == allele_383650.model_dump()
+
+
+def test_class_refatt_map():
+    class_refatt_map_expected = {
+        'Allele': ['location'],
+        'Haplotype': ['members'],
+        '_CopyNumber': ['location'],
+        'CopyNumberCount': ['location'],
+        'CopyNumberChange': ['location'],
+        'GenotypeMember': ['variation']
+    }
+    assert class_refatt_map_expected == models.class_refatt_map