engine_tests.py 文件源码

def make_test_data():
    from sklearn.datasets import make_classification
    import pandas as pd

    data = make_classification(n_samples=3, n_features=4)
    data = data[0]
    df = pd.DataFrame(data, columns=list("ABCD"))

    prepare_path(test_data_file)
    df.to_csv(test_data_file, sep='\t', index=False)