fix tests

NPLinker · Sep 24, 2024 · a74d8b5 · a74d8b5
1 parent 70ccd5e
commit a74d8b5
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/test_callbacks.py b/tests/test_callbacks.py
@@ -44,13 +44,15 @@ def sample_processed_data():
                 "# BGCs": 3,
                 "BGC Classes": ["NRPS", "PKS"],
                 "BGC IDs": ["BGC_1", "BGC_2", "BGC_3"],
+                "BGC smiles": ["CCO", "CCN", "N/A"],
                 "strains": ["Strain_1", "Strain_2", "Strain_3"],
             },
             {
                 "GCF ID": "GCF_2",
                 "# BGCs": 2,
                 "BGC Classes": ["RiPP", "Terpene"],
                 "BGC IDs": ["BGC_1", "BGC_3"],
+                "BGC smiles": ["CCO", "N/A"],
                 "strains": ["Strain_3"],
             },
         ]