Merge pull request #263 from ga4gh/2a-class-refatt-test

Add a test for the class refatt map to ensure model changes don't bre…
ga4gh · Sep 18, 2023 · b80c4af · b80c4af
2 parents a635b1f + 6ede345
commit b80c4af
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