def test_sort_dir_map(self, galaxy, class_, expected):
maps = Maps(plateifu=galaxy.plateifu)
ha = maps['emline_gflux_ha_6564']
dir_ = _sort_dir(ha, class_)
dir_public = [it for it in dir_ if it[0] != '_']
assert set(dir_public) == set(expected)
def test_sort_dir_spectrum(self, galaxy, class_, expected):
cube = Cube(plateifu=galaxy.plateifu)
spax = cube[0, 0]
spec = spax.flux
dir_ = _sort_dir(spec, class_)
dir_public = [it for it in dir_ if it[0] != '_']
assert set(dir_public) == set(expected)