def sample_ts_files(size, labels=[None]): temp_dir = tempfile.mkdtemp() paths = [] for label in islice(cycle(labels), size): t, m, e = sample_values() name = str(uuid.uuid4()) path = pjoin(temp_dir, '{}.npz'.format(name)) ts = TimeSeries(t, m, e, label=label, path=path, name=name) ts.save(path) paths.append(path) yield paths shutil.rmtree(temp_dir)
def sample_ts_files(size, labels=[None]): temp_dir = tempfile.mkdtemp() paths = [] for label in islice(cycle(labels), size): t, m, e = sample_values() name = str(uuid.uuid4()) path = pjoin(temp_dir, '{}.npz'.format(name)) ts = TimeSeries(t, m, e, label=label, path=path, name=name) ts.save(path) paths.append(path) yield paths shutil.rmtree(temp_dir)
def test_time_series_npz(): n_channels = 3 t, m, e = sample_time_series(channels=n_channels) ts = TimeSeries(t[0], m[0], e[0]) ts.save(TEST_TS_PATH) ts_loaded = time_series.load(TEST_TS_PATH) assert_ts_equal(ts, ts_loaded) ts = TimeSeries(t[0], m, e[0]) ts.save(TEST_TS_PATH) ts_loaded = time_series.load(TEST_TS_PATH) assert_ts_equal(ts, ts_loaded) t = [t[i][0:i+2] for i in range(len(t))] m = [m[i][0:i+2] for i in range(len(m))] e = [e[i][0:i+2] for i in range(len(e))] ts = TimeSeries(t, m, e) ts.save(TEST_TS_PATH) ts_loaded = time_series.load(TEST_TS_PATH) assert_ts_equal(ts, ts_loaded)
def test_time_series_npz(tmpdir): n_channels = 3 t, m, e = sample_time_series(channels=n_channels) ts = TimeSeries(t[0], m[0], e[0]) ts_path = os.path.join(str(tmpdir), str(uuid4()) + '.npz') ts.save(ts_path) ts_loaded = time_series.load(ts_path) assert_ts_equal(ts, ts_loaded) ts = TimeSeries(t[0], m, e[0]) ts_path = os.path.join(str(tmpdir), str(uuid4()) + '.npz') ts.save(ts_path) ts_loaded = time_series.load(ts_path) assert_ts_equal(ts, ts_loaded) t = [t[i][0:i + 2] for i in range(len(t))] m = [m[i][0:i + 2] for i in range(len(m))] e = [e[i][0:i + 2] for i in range(len(e))] ts = TimeSeries(t, m, e) ts_path = os.path.join(str(tmpdir), str(uuid4()) + '.npz') ts.save(ts_path) ts_loaded = time_series.load(ts_path) assert_ts_equal(ts, ts_loaded)
def test_time_series_npz(tmpdir): n_channels = 3 t, m, e = sample_time_series(channels=n_channels) ts = TimeSeries(t[0], m[0], e[0]) ts_path = os.path.join(str(tmpdir), str(uuid4()) + '.npz') ts.save(ts_path) ts_loaded = time_series.load(ts_path) assert_ts_equal(ts, ts_loaded) ts = TimeSeries(t[0], m, e[0]) ts_path = os.path.join(str(tmpdir), str(uuid4()) + '.npz') ts.save(ts_path) ts_loaded = time_series.load(ts_path) assert_ts_equal(ts, ts_loaded) t = [t[i][0:i+2] for i in range(len(t))] m = [m[i][0:i+2] for i in range(len(m))] e = [e[i][0:i+2] for i in range(len(e))] ts = TimeSeries(t, m, e) ts_path = os.path.join(str(tmpdir), str(uuid4()) + '.npz') ts.save(ts_path) ts_loaded = time_series.load(ts_path) assert_ts_equal(ts, ts_loaded)