def test_shipper_cant_reregister(): register_shipper('myshipper', MyShipper) assert_raises(RuntimeError, register_shipper, 'myshipper', MyShipper) unregister_shipper('myshipper')
def test_shipper_registration(): register_shipper('myshipper', MyShipper) assert_equal(get_shipper('myshipper'), MyShipper) unregister_shipper('myshipper') assert_equal(get_shipper('myshipper'), None)