def test_docker_extract_features():
"""Test main Docker extract features method"""
script_fpath = pjoin(DATA_PATH, "testfeature1.py")
ts_datafile = pjoin(DATA_PATH, "dotastro_215153.dat")
t, m, e = ctt.parse_ts_data(ts_datafile)
feats_known_dict = {'t': t, 'm': m, 'e': e}
results = cft.docker_extract_features(script_fpath, feats_known_dict)
assert(isinstance(results, dict))
npt.assert_almost_equal(results["avg_mag"], 10.347417647058824)
def test_docker_extract_features():
"""Test main Docker extract features method"""
script_fpath = pjoin(DATA_PATH, "testfeature1.py")
ts_datafile_paths = [pjoin(DATA_PATH, "dotastro_215153.dat")]
results = cft.docker_extract_features(script_fpath,
ts_datafile_paths=ts_datafile_paths)
npt.assert_equal(len(results), 1)
assert(isinstance(results[0], dict))
npt.assert_almost_equal(results[0]["avg_mag"], 10.347417647058824)
