def test_dataprep(self): config = Config() assert not TEST_IMAGE_DIR_RES.exists() assert not Path(TEST_JOB_DIR / 'class_mapping.json').exists() assert not Path(TEST_JOB_DIR / 'test_samples.json').exists() assert not Path(TEST_JOB_DIR / 'train_samples.json').exists() assert not Path(TEST_JOB_DIR / 'val_samples.json').exists() dataprep(config, config_file=TEST_CONFIG_DATAPREP) assert config.dataprep['run'] == True assert config.dataprep['job_dir'] == str(TEST_JOB_DIR) assert config.dataprep['samples_file'] == str(TEST_SAMPLES) assert config.dataprep['image_dir'] == str(TEST_IMAGE_DIR) assert config.dataprep['resize'] == True assert config.train['run'] == False assert config.evaluate['run'] == False assert config.cloud['run'] == False assert TEST_IMAGE_DIR_RES.exists() assert Path(TEST_JOB_DIR / 'class_mapping.json').exists() assert Path(TEST_JOB_DIR / 'test_samples.json').exists() assert Path(TEST_JOB_DIR / 'train_samples.json').exists() assert Path(TEST_JOB_DIR / 'val_samples.json').exists()
def dataprep(config: Config, **kwargs): """Run data preparation and create job dir. Creates a directory (job_dir) with the following files: - train_samples.json - val_samples.json - test_samples.json - class_mapping.json """ commands.dataprep(config, **kwargs)