test_utils.py 文件源码

def test_modelmethod():
    session = Session()

    with pytest.raises(ValueError) as e:
        ModelStub.my_model_method(1, 2)

    c, s, a = ModelStub.my_model_method(session, 1)

    assert c is ModelStub
    assert s is session
    assert a == 1