def test_parse_metadata_file(): """Test parse metadata file.""" meta_feats = pred.parse_metadata_file( pjoin(DATA_PATH, "215153_215176_218272_218934_metadata.dat")) assert("dotastro_215153.dat" in meta_feats) assert("meta1" in meta_feats["dotastro_215153.dat"]) npt.assert_almost_equal(meta_feats["dotastro_215153.dat"]["meta1"], 0.23423)
def test_featurize_single(): """Test featurization of single TS data file""" meta_feats = pred.parse_metadata_file( pjoin(DATA_PATH, "215153_metadata.dat")) res_dict = pred.featurize_single( pjoin(DATA_PATH, "dotastro_215153.dat"), ["std_err"], pjoin(DATA_PATH, "testfeature1.py"), meta_feats) assert all("std_err" in d["features_dict"] for fname, d in res_dict.items()) assert all("ts_data" in d for fname, d in res_dict.items())
def test_featurize_multiple_serially(): """Test serial featurization""" meta_feats = pred.parse_metadata_file( pjoin(DATA_PATH, "215153_215176_218272_218934_metadata.dat")) res_dict = pred.featurize_multiple_serially( pjoin(DATA_PATH, "215153_215176_218272_218934.tar.gz"), "/tmp", ["std_err"], pjoin(DATA_PATH, "testfeature1.py"), meta_feats) npt.assert_equal(len(res_dict), 4) assert all("std_err" in d["features_dict"] for fname, d in res_dict.items()) assert all("ts_data" in d for fname, d in res_dict.items())