def test_featurize_tsdata_object(): """Test featurize TS data object function""" path_to_csv = pjoin(DATA_PATH, "dotastro_215153.dat") short_fname = featurize.shorten_fname(path_to_csv) custom_script_path = pjoin(cfg.UPLOAD_FOLDER, "testfeature1.py") fname_class_dict = {"dotastro_215153": "Mira"} features_to_use = ["std_err", "freq1_harmonics_freq_0"] all_feats = featurize.featurize_tsdata_object( path_to_csv, short_fname, custom_script_path, fname_class_dict, features_to_use) assert(isinstance(all_feats, dict)) assert("std_err" in all_feats) assert("freq1_harmonics_freq_0" in all_feats)
def test_shorten_fname(): """Test shorten_fname.""" npt.assert_equal(featurize.shorten_fname("path/to/filename.sfx"), "filename") npt.assert_equal(featurize.shorten_fname("/home/path/abc.dat"), "abc")