Python Order.del_main示例

示例#1

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)

示例#2

def test_del_main_correct_input():
    order = Order(1)
    main = order.add_main('burger')
    order.del_main(main)
    assert not order.mains