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)