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"])