def test_hdf5writer_dict():
f = Frame()
f.n = {1: 1}
f.writer = writers.hdf5writer()
with pytest.raises(NotImplementedError):
f.writeoutput(0)
shutil.rmtree(f.writer.datadir)
def test_hdf5writer_skip():
f = Frame()
f.writer = writers.hdf5writer()
f.addfield("x", 0., save=False)
f.writeoutput(0)
with pytest.raises(KeyError):
x = f.writer.read.sequence("x")
shutil.rmtree(f.writer.datadir)
def test_hdf5writer_list():
f = Frame()
f.n = [1, 1]
f.writer = writers.hdf5writer()
f.writeoutput(0)
f.writeoutput(1)
data0000 = f.writer.read.output(0)
assert np.all(data0000.n == [1, 1])
n = f.writer.read.sequence("n")
assert np.all(n == [[1, 1], [1, 1]])
data = f.writer.read.all()
assert np.all(data.n == [[1, 1], [1, 1]])
shutil.rmtree(f.writer.datadir)
def test_hdf5writer_single_value_array():
f = Frame()
f.n = np.array(1)
f.writer = writers.hdf5writer()
f.writeoutput(0)
f.writeoutput(1)
data0000 = f.writer.read.output(0)
assert data0000.n == 1
n = f.writer.read.sequence("n")
assert np.all(n == [1, 1])
data = f.writer.read.all()
assert np.all(data.n == [1, 1])
shutil.rmtree(f.writer.datadir)
def test_hdf5writer_strings():
string = "test"
# When read from HDF5 the string will be a byte literal
string_cmpr = string.encode()
f = Frame()
f.addfield("s", string)
f.t = string
f.writer = writers.hdf5writer()
f.writeoutput(0)
f.writeoutput(1)
data0000 = f.writer.read.output(0)
assert data0000.s[0] == string_cmpr
assert data0000.t == string_cmpr
s = f.writer.read.sequence("s")
assert np.all(s == [string_cmpr, string_cmpr])
t = f.writer.read.sequence("t")
assert np.all(t == [string_cmpr, string_cmpr])
data = f.writer.read.all()
assert np.all(data.s == [string_cmpr, string_cmpr])
assert np.all(data.s == [string_cmpr, string_cmpr])
shutil.rmtree(f.writer.datadir)
def test_hdf5writer_none():
f = Frame()
f.n = None
f.writer = writers.hdf5writer()
with pytest.raises(ValueError):
f.writeoutput(0)
f.n = [1, None]
with pytest.raises(ValueError):
f.writeoutput(1)
f.n = (1, None)
with pytest.raises(ValueError):
f.writeoutput(2)
shutil.rmtree(f.writer.datadir)
