Source code for kitcar_ml.onboarding.test.test_setup

import os


[docs]def test_dataset_download():
    print("Test Setup script")

    path_onboarding = os.path.join(os.path.dirname(__file__), "..")

    os.system(f'python3 {os.path.join(path_onboarding, "setup.py")}')

    assert os.path.exists(
        os.path.join(path_onboarding, ".dataset.zip")
    ), "Setup script did not download the zipped dataset."

    files = next(os.walk(os.path.join(path_onboarding, "dataset/")))[2]
    assert len(files) >= 1000, "Setup script did not download the dataset."


[docs]def main():
    print("Test Onboarding Setup")
    test_dataset_download()


if __name__ == "__main__":
    main()