def test_spin_up_and_run_container():
"""Test spin up and run container"""
client, cont_id = ridc.spin_up_and_run_container("featurize", "/tmp")
conts = client.containers(all=True)
found_match = False
for cont in conts:
if cont["Id"] == cont_id:
found_match = True
break
client.remove_container(container=cont_id, force=True)
assert found_match
def test_copy_results_files_featurize():
"""Test copy results files - featurize"""
copied_data_dir = os.path.join(cfg.PROJECT_PATH, "tmp",
str(uuid.uuid4())[:10])
os.makedirs(copied_data_dir)
featurize_setup()
shutil.copy(
pjoin(DATA_PATH, "testfeature1.py"),
cfg.CUSTOM_FEATURE_SCRIPT_FOLDER)
headerfile_path = pjoin(
cfg.UPLOAD_FOLDER, "asas_training_subset_classes_with_metadata.dat")
zipfile_path = pjoin(cfg.UPLOAD_FOLDER,
"asas_training_subset.tar.gz")
features_to_use = ["std_err", "freq1_harmonics_freq_0"]
featureset_key = "TEST01"
is_test = True
already_featurized = False
custom_script_path = pjoin(cfg.CUSTOM_FEATURE_SCRIPT_FOLDER,
"testfeature1.py")
args_dict = locals()
tmp_files = ridc.copy_data_files_featurize_prep(args_dict)
client, cont_id = ridc.spin_up_and_run_container("featurize",
copied_data_dir)
ridc.copy_results_files_featurize(featureset_key, client, cont_id)
shutil.rmtree(copied_data_dir, ignore_errors=True)
assert(os.path.exists(pjoin(cfg.FEATURES_FOLDER,
"TEST01_features.csv")))
assert(os.path.exists(pjoin(cfg.FEATURES_FOLDER,
"TEST01_classes.npy")))
assert(os.path.exists(pjoin(pjoin(cfg.MLTSP_PACKAGE_PATH,
"Flask/static/data"),
"TEST01_features_with_classes.csv")))
os.remove(pjoin(cfg.FEATURES_FOLDER, "TEST01_classes.npy"))
os.remove(pjoin(cfg.FEATURES_FOLDER, "TEST01_features.csv"))
os.remove(pjoin(pjoin(cfg.MLTSP_PACKAGE_PATH,
"Flask/static/data"),
"TEST01_features_with_classes.csv"))
featurize_teardown()
