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())
