返回详细assert失败信息

原代码摘自StackOverflow
如果有多条assert信息，发生错误如何定位？ 例如：

try:
    assert True
    assert 7 == 7
    assert 1 == 2
    # many more statements like this
except AssertionError:
    print 'Houston, we have a problem.'
    print
    print 'An error occurred on line ???? in statement ???'
    exit(1)

其中一种方法是写多条assert语句，但是太繁琐。

通过导入traceback模块，可以获得详尽的信息：

except AssertionError:
    _, _, tb = sys.exc_info()
    traceback.print_tb(tb) # Fixed format
    tb_info = traceback.extract_tb(tb)
    filename, line, func, text = tb_info[-1]

    print('An error occurred on line {} in statement {}'.format(line, text))
    exit(1)