def test_main_predict_tarball_cust_feats(): """Test main predict function - tarball w/ custom feats""" generate_model_cust_feats() shutil.copy(pjoin(DATA_PATH, "215153_215176_218272_218934.tar.gz"), cfg.UPLOAD_FOLDER) shutil.copy(pjoin(DATA_PATH, "215153_215176_218272_218934_metadata.dat"), cfg.UPLOAD_FOLDER) shutil.copy(pjoin(DATA_PATH, "testfeature1.py"), pjoin(cfg.CUSTOM_FEATURE_SCRIPT_FOLDER, "TESTRUN_CF.py")) pred_results_dict = pred.predict( pjoin(cfg.UPLOAD_FOLDER, "215153_215176_218272_218934.tar.gz"), "TEMP_TEST01", "RF", "TEMP_TEST01", metadata_file_path=pjoin(cfg.UPLOAD_FOLDER, "215153_215176_218272_218934_metadata.dat"), custom_features_script=pjoin(cfg.CUSTOM_FEATURE_SCRIPT_FOLDER, "TESTRUN_CF.py")) os.remove(pjoin(cfg.UPLOAD_FOLDER, "215153_215176_218272_218934.tar.gz")) os.remove(pjoin(cfg.UPLOAD_FOLDER, "215153_215176_218272_218934_metadata.dat")) os.remove(pjoin(cfg.FEATURES_FOLDER, "TEMP_TEST01_features.csv")) os.remove(pjoin(cfg.FEATURES_FOLDER, "TEMP_TEST01_classes.npy")) os.remove(pjoin(cfg.MODELS_FOLDER, "TEMP_TEST01_RF.pkl")) os.remove(pjoin(cfg.CUSTOM_FEATURE_SCRIPT_FOLDER, "TESTRUN_CF.py")) npt.assert_equal( len(pred_results_dict["dotastro_215153"]["pred_results_list"]), 3) npt.assert_equal(len(pred_results_dict), 4) for res_dict in pred_results_dict.values(): assert(all(el[0] in ['Mira', 'Herbig_AEBE', 'Beta_Lyrae', 'Classical_Cepheid', 'W_Ursae_Maj', 'Delta_Scuti'] for el in res_dict["pred_results_list"]))
def test_main_predict(): """Test main predict function""" generate_model() shutil.copy(pjoin(DATA_PATH, "dotastro_215153.dat"), pjoin(cfg.UPLOAD_FOLDER, "TESTRUN_215153.dat")) shutil.copy(pjoin(DATA_PATH, "TESTRUN_215153_metadata.dat"), cfg.UPLOAD_FOLDER) pred_results_dict = pred.predict( pjoin(cfg.UPLOAD_FOLDER, "TESTRUN_215153.dat"), "TEMP_TEST01", "RF", "TEMP_TEST01", metadata_file_path=pjoin(cfg.UPLOAD_FOLDER, "TESTRUN_215153_metadata.dat"), custom_features_script=None) os.remove(pjoin(cfg.UPLOAD_FOLDER, "TESTRUN_215153.dat")) os.remove(pjoin(cfg.UPLOAD_FOLDER, "TESTRUN_215153_metadata.dat")) os.remove(pjoin(cfg.FEATURES_FOLDER, "TEMP_TEST01_features.csv")) os.remove(pjoin(cfg.FEATURES_FOLDER, "TEMP_TEST01_classes.npy")) os.remove(pjoin(cfg.MODELS_FOLDER, "TEMP_TEST01_RF.pkl")) npt.assert_equal( len(pred_results_dict["TESTRUN_215153"]["pred_results_list"]), 3) assert(all(all(el[0] in ['Mira', 'Herbig_AEBE', 'Beta_Lyrae', 'Classical_Cepheid', 'W_Ursae_Maj', 'Delta_Scuti'] for el in pred_results_dict[fname]\ ["pred_results_list"]) for fname in pred_results_dict))
def test_predict(): """Test class prediction.""" results_dict = predict_class.predict( pjoin(DATA_PATH, "dotastro_215153.dat"), "TESTRUN", "RF", "TESTRUN", custom_features_script=pjoin(cfg.CUSTOM_FEATURE_SCRIPT_FOLDER, "testfeature1.py"), metadata_file_path=pjoin(DATA_PATH, "215153_metadata.dat"), ) assert isinstance(results_dict, dict) assert "dotastro_215153.dat" in results_dict assert isinstance(results_dict["dotastro_215153.dat"]["features_dict"]["std_err"], float)