test_util.py 文件源码

def verify_result(result, expected, float_cmp, debug):
    if not isinstance(result, pd.core.series.Series):
        if result == expected:
            print('TEST OK')
            return
        else:
            print('TEST Failed.')
            return
    result = result.dropna()
    expected = expected.dropna()
    if debug:
        print('RESULT:')
        print(result)
        print('EXPECTED:')
        print(expected)
    if float_cmp:
        cmp = (np.abs(result - expected) < 2.631048e-06)
    else:
        cmp = (result == expected)
    if len(cmp[cmp == False]) > 0 :
        print('TEST Failed.')
        return
    print('TEST OK.')
    return