python: assert testing 笔记

from xxx(.py文件) import xxx(def名称)

 #test functions in the new program

def two_output():
    expected = (2.5, 1.0)
    actual = quadratic(2, -7, 5)
    assert actual == expected, "Test 1 : failed!"

expected = 期望的输出结果；actual = 运行function后的实际结果

assert actual == expected，结果为Boolean。若Boolean为False，则print “Test 1 : failed!”

=========================================================================

def no_real_solution():
    expected = ()
    actual = quadratic(2,4,5)
    assert actual == expected, "Test 3: failed!"
    print('no_real_solution test passed')

若assert结果为True，则print “no_real_solution test passes”. (前提是no real solution后function直接return None)

=========================================================================

def bad_input_type():
    try:
        quadratic(2.5,4,1)
        #test fail
        assert False, "No exception raised"
    except TypeError:
        #test should pass
        print("Test bad_input_type passed")
    except Exception:
        #fail
        assert False, "Test bad_input_type failed: wrong exception raised!"

当期待function报错时，使用try和except进行检测。

1. try：function不报错则assert False，输出对应结论"No exception raised".

2. except TypeError: 假设期待报错TypeError，则test pass，输出结论"test passed".

3. except Exception: 假设报错，但错误类型不正确，assert False，输出结论"wrong exception rasied."

=========================================================================

运行代码：

if __name__ == "__main__":
    two_input()
    no_real_solution()
    bad_input_type()