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
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
