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