def test_determine_feats_to_plot1():
"""Test determine feats to plot - 1"""
ftp = featurize.determine_feats_to_plot(["abc", "221a", "22d"])
assert("221a" in ftp)
assert("median" not in ftp)
ftp = featurize.determine_feats_to_plot(cfg.features_list_science)
assert("freq1_amplitude1" in ftp)
assert("median" in ftp)
def test_determine_feats_to_plot2():
"""Test determine features to plot - 2"""
ftpl = featurize.determine_feats_to_plot(cfg.features_to_plot)
npt.assert_array_equal(ftpl, cfg.features_to_plot)
ftpl = featurize.determine_feats_to_plot(cfg.features_list_science)
npt.assert_array_equal(ftpl, cfg.features_to_plot)
ftpl = featurize.determine_feats_to_plot(cfg.features_list_obs)
npt.assert_equal(len(ftpl), 5)
assert(all(f in cfg.features_list_obs for f in ftpl))
ftpl = featurize.determine_feats_to_plot(["f1"])
npt.assert_array_equal(ftpl, ["f1"])
