def test_subset_select(self): """select multiple columns.""" timeseries = TimeSeries(IN_OUT_DATA) kcol = (Pipeline().from_source(timeseries).select( ['out', 'perpendicular']).to_keyed_collections()) new_ts = TimeSeries( dict(name='new_timeseries', collection=kcol.get('all'))) self.assertEqual(set(new_ts.columns()), set(['out', 'perpendicular']))
def test_single_select(self): """select a single column.""" timeseries = TimeSeries(IN_OUT_DATA) kcol = (Pipeline().from_source(timeseries).select( 'in').to_keyed_collections()) new_ts = TimeSeries( dict(name='new_timeseries', collection=kcol.get('all'))) self.assertEqual(new_ts.columns(), ['in'])
def test_subset_select(self): """select multiple columns.""" timeseries = TimeSeries(IN_OUT_DATA) kcol = ( Pipeline() .from_source(timeseries) .select(['out', 'perpendicular']) .to_keyed_collections() ) new_ts = TimeSeries(dict(name='new_timeseries', collection=kcol.get('all'))) self.assertEqual(set(new_ts.columns()), set(['out', 'perpendicular']))
def test_single_select(self): """select a single column.""" timeseries = TimeSeries(IN_OUT_DATA) kcol = ( Pipeline() .from_source(timeseries) .select('in') .to_keyed_collections() ) new_ts = TimeSeries(dict(name='new_timeseries', collection=kcol.get('all'))) self.assertEqual(new_ts.columns(), ['in'])