ugh

cancervariants · Nov 17, 2023 · 272ea4b · 272ea4b
1 parent 832068d
commit 272ea4b
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/unit/test_disease_indication.py b/tests/unit/test_disease_indication.py
@@ -25,6 +25,8 @@ def test_normalize_disease(is_test_env: bool, db: Database):
     # set up normalizer
     os.environ["DISEASE_DYNAMO_TABLE"] = "disease_normalizer_therapy_test"
     chembl = ChEMBL(db)
+    chembl.disease_normalizer.db.drop_db()
+    chembl.disease_normalizer.db.initialize_db()
     chembl.disease_normalizer.db.add_source_metadata(
         DiseaseSourceName.MONDO,
         DiseaseSourceMeta(
@@ -47,5 +49,6 @@ def test_normalize_disease(is_test_env: bool, db: Database):
         },
         DiseaseSourceName.MONDO,
     )
+    chembl.disease_normalizer.db.complete_write_transaction()
     result = chembl._normalize_disease("mondo:0700110")
     assert result == "mondo:0700110"