예제 #1
0
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)
예제 #2
0
파일: fixtures.py 프로젝트: acrellin/cesium
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)
예제 #3
0
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)
예제 #4
0
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)
예제 #5
0
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)