def test_hdf5_write_with_properties(tmp_path):
"""Test writing an HDF5 file with some objects with additional properties."""
fn = os.path.join(tmp_path, "test.h5")
objects = []
for i in range(10):
obj, _ = create_test_object(id=i)
obj.properties = create_test_properties()
objects.append(obj)
with btrack.dataio.HDF5FileHandler(fn, 'w') as h:
h.write_objects(objects)
# now try to read those objects and compare with those used to write
with btrack.dataio.HDF5FileHandler(fn, 'r') as h:
objects_from_file = h.objects
extra_props = list(create_test_properties().keys())
properties = ['x', 'y', 'z', 't', 'label', 'ID']
for orig, read in zip(objects, objects_from_file):
for p in properties:
# use all close, since h5 file stores in float32 default
np.testing.assert_allclose(getattr(orig, p), getattr(read, p))
for p in extra_props:
np.testing.assert_allclose(orig.properties[p], read.properties[p])
def test_object():
"""Test that an object is correctly instantiated, and that the stored
data matches the data used for creation."""
obj, data = create_test_object()
for k, v in data.items():
assert getattr(obj, k) == v
def test_object_properties(properties: dict):
"""Test an object with some properties."""
obj, data = create_test_object()
obj.properties = properties
for k, v in properties.items():
assert obj.properties[k] == v