def test_filter_df(tinyds): datasets_df = tinyds.to_df() # Test filtering on different Instruments filtered_df = utils.filter_df(datasets_df, instrument="bassoon") assert filtered_df["instrument"].unique() == ["bassoon"] filtered_df = utils.filter_df(datasets_df, instrument="trumpet") assert filtered_df["instrument"].unique() == ["trumpet"] # Test filtering on different datasets and dataset combinations. filtered_df = utils.filter_df(datasets_df, datasets=["rwc"]) assert filtered_df["dataset"].unique() == ["rwc"] filtered_df = utils.filter_df(datasets_df, datasets=["philharmonia"]) assert filtered_df["dataset"].unique() == ["philharmonia"] filtered_df = utils.filter_df(datasets_df, datasets=["philharmonia", "rwc"]) assert set(filtered_df["dataset"].unique()) == \ set(["philharmonia", "rwc"])
def test_filter_df(tinyds): datasets_df = tinyds.to_df() # Test filtering on different Instruments filtered_df = utils.filter_df( datasets_df, instrument="bassoon") assert filtered_df["instrument"].unique() == ["bassoon"] filtered_df = utils.filter_df( datasets_df, instrument="trumpet") assert filtered_df["instrument"].unique() == ["trumpet"] # Test filtering on different datasets and dataset combinations. filtered_df = utils.filter_df( datasets_df, datasets=["rwc"]) assert filtered_df["dataset"].unique() == ["rwc"] filtered_df = utils.filter_df( datasets_df, datasets=["philharmonia"]) assert filtered_df["dataset"].unique() == ["philharmonia"] filtered_df = utils.filter_df( datasets_df, datasets=["philharmonia", "rwc"]) assert set(filtered_df["dataset"].unique()) == \ set(["philharmonia", "rwc"])
def _instrument_streams(self, instrument): """Return a list of generators for all records in the dataframe which match the instrument given, and are in the datasets. Parameters ---------- instrument : str Instrument to select. Returns ------- streams : list of pescador.Streamer One streamer for each instrument file. """ # Get list of instruments instrument_records = utils.filter_df(self.features_df, instrument=instrument) seed_pool = [ pescador.Streamer(self.record_slicer, record, self.t_len, **self.slicer_kwargs) for _, record in instrument_records.iterrows() ] return seed_pool