def test_check_product(): """ Tests api check_product """ entities = [e,e1] a = Api(entities) assert a.check_product(p3) == ['EPA heartily approves of Zoobiedoo', 'Moo Moo Meadows heartily approves of Zoobiedoo'] assert a.check_product(p) == ['EPA heartily approves of Green Soap', 'Moo Moo Meadows thinks that Green Soap is super lame!!!']
def test_add_entity(): """ Tests Api add_entity """ entities = [e,e1] a = Api(entities) a.add_entity(e2) assert a.entities == [e,e1,e2]