def test_memmap():
with TempDirectory() as directory:
io = npy.NumpyDirectory(directory.path, memmap=True)
original = utilities.fake_single_stream()
name = 'stream'
io.write(original, name)
assert original == io.read(name)
def test_read_write_stream():
with TempDirectory() as directory:
filename = 'test.nc'
io = nc.NCFile(os.path.join(directory.path, filename))
original = utilities.fake_single_stream()
name = 'stream'
io.write(original, name)
assert original == io.read(name)
def test_cached_single_stream():
with TempDirectory() as directory:
filename = 'test.nc'
io = nc.NCFile(os.path.join(directory.path, filename), cache_s21_raw=True)
original = utilities.fake_single_stream()
name = 'measurement'
io.write(original, name)
assert np.all(original.s21_raw == io.read(name).s21_raw)
def test_read_write_stream():
with TempDirectory() as directory:
filename = 'test.nc'
io = nc.NCFile(os.path.join(directory.path, filename))
original = utilities.fake_single_stream()
name = 'stream'
io.write(original, name)
assert original == io.read(name)
def test_eq_array():
m1 = utilities.fake_single_stream()
m2 = basic.SingleStream(**dict([(k, copy.copy(v)) for k, v in m1.__dict__.items() if not k.startswith('_')]))
assert m1 == m2
index = np.random.random_integers(0, m1.s21_raw.size)
m2.s21_raw[index] += 1
assert m1 != m2
m1.s21_raw[index] = m2.s21_raw[index] = np.nan
assert m1 == m2
def test_cached_single_stream():
with TempDirectory() as directory:
filename = 'test.nc'
io = nc.NCFile(os.path.join(directory.path, filename),
cache_s21_raw=True)
original = utilities.fake_single_stream()
name = 'measurement'
io.write(original, name)
assert np.all(original.s21_raw == io.read(name).s21_raw)
def test_eq_array():
m1 = utilities.fake_single_stream()
m2 = basic.SingleStream(**dict([(k, copy.copy(v))
for k, v in m1.__dict__.items()
if not k.startswith('_')]))
assert m1 == m2
index = np.random.random_integers(0, m1.s21_raw.size)
m2.s21_raw[index] += 1
assert m1 != m2
m1.s21_raw[index] = m2.s21_raw[index] = np.nan
assert m1 == m2
