From 489b93b4c242660a804aa8f790e6ac9f5930c133 Mon Sep 17 00:00:00 2001 From: Thomas Wood Date: Wed, 21 Aug 2024 05:56:00 +0100 Subject: [PATCH] add new test case --- tests/test_drugs_finder.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/tests/test_drugs_finder.py b/tests/test_drugs_finder.py index 84f917b..c21643b 100644 --- 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?