def test_del_main_incorrect_input(): order = Order(1) order.add_main('burger') wrong_main = Main('wrap') with pytest.raises(DevError) as e: order.del_main(wrong_main) assert "main that you're trying to delete doesn't exist in this order" in str(e.value)
def test_del_main_correct_input(): order = Order(1) main = order.add_main('burger') order.del_main(main) assert not order.mains