def test_part_2_lookup_pharm_classes_warfarin(self): # TODO: Complete test by adding call to method to look up pharm classes for warfarin endpoint = Endpoint("mock") pharm_classes = endpoint.get_pharm_class("warfarin") expected_mappings = { 'EPC': ['Vitamin K Antagonist'], 'MoA': ['Vitamin K Inhibitors'] } self.assertDictEqual(pharm_classes, expected_mappings)
def test_part_2_lookup_pharm_classes_aspirin(self): # TODO: Complete test by adding call to method to look up pharm classes for aspirin endpoint = Endpoint("mock") pharm_classes = endpoint.get_pharm_class("aspirin") expected_mappings = { 'Chemical/Ingredient': ['Nonsteroidal Anti-inflammatory Compounds'], 'PE': [ 'Decreased Prostaglandin Production', 'Decreased Platelet Aggregation' ], 'EPC': [ 'Nonsteroidal Anti-inflammatory Drug', 'Platelet Aggregation Inhibitor' ], 'MoA': ['Cyclooxygenase Inhibitors'] } self.assertDictEqual(pharm_classes, expected_mappings)
def test_get_pharm_class(self): endpoint = Endpoint("fda") pharm_class = endpoint.get_pharm_class("warfarin") print pharm_class