add new test case

fastdatascience · Aug 21, 2024 · 489b93b · 489b93b
1 parent c5622d3
commit 489b93b
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/tests/test_drugs_finder.py b/tests/test_drugs_finder.py
@@ -142,3 +142,8 @@ def test_structure_2(self):
         self.assertEqual(1, len(drugs))
         self.assertEqual("Guaifenesin", drugs[0][0]['name'])
         self.assertIn("0.0000 C", drugs[0][0]['structure_mol'])
+
+    def test_penicillin_streptomycin(self):
+        drugs = find_drugs("i bought some Penicillin streptomycin".split(" "), is_include_structure=True)
+
+        self.assertEqual(2, len(drugs))  # should be 1?