def test_compare_copy_default_format(snapshot_path):
""""""
with Snapshot(snapshot_path) as snap:
new_snap_path = snapshot_path + "-test-copy"
with Snapshot(new_snap_path) as new_snap:
assert new_snap.format is FileFormat.DEFAULT
assert new_snap.header == snap.header
print(new_snap.inspect())
def test_copy(snapshot_path):
""""""
with Snapshot(snapshot_path) as snap:
new_snap_path = snapshot_path + "-test-copy"
with Snapshot(new_snap_path, "w") as new_snap:
# First assign the header.
new_snap.header = snap.header
# Block assignment order matters.
new_snap["POS"] = snap["POS"]
new_snap["VEL"] = snap["VEL"]
new_snap["ID"] = snap["ID"]
new_snap.flush()
def test_copy_block_twice(snapshot_path):
""""""
with pytest.raises(KeyError):
with Snapshot(snapshot_path) as snap:
new_snap_path = snapshot_path + "-test-snap"
with Snapshot(new_snap_path, "w") as new_snap:
# First assign the header.
new_snap.header = snap.header
# This must fail.
new_snap["POS"] = snap["POS"]
new_snap.flush()
new_snap["POS"] = snap["POS"]
new_snap.flush()
def test_copy_default_format(snapshot_path):
""""""
with Snapshot(snapshot_path) as snap:
new_snap_path = snapshot_path + "-test-copy"
with Snapshot(new_snap_path, "w",
format=FileFormat.DEFAULT) as new_snap:
# First assign the header.
new_snap.header = snap.header
# Block assignment order matters.
new_snap["POS"] = snap["POS"]
new_snap["VEL"] = snap["VEL"]
new_snap["ID"] = snap["ID"]
new_snap.flush()
def test_iter(snapshot_path):
""""""
with Snapshot(snapshot_path) as snap:
for block in snap.values():
print("************ Block ************")
print(block)
print("Number of reachable snapshot blocks: {}".format(len(snap)))
def test_read_invalid_mode(snapshot_path):
""""""
with pytest.raises(ValueError):
# Although snapshots are open as binary, the mode does not accept
# the ``b`` modifier.
with Snapshot(snapshot_path, "rb") as snap:
print(snap)
def test_load_blocks(snapshot_path):
""""""
with Snapshot(snapshot_path) as snap:
positions1 = snap["POS"]
velocities = snap["VEL"]
positions2 = snap["POS"]
print(snap.header)
print(positions1)
print(positions2)
print(velocities)
def test_create(dummy_snap_header, dummy_snap_block):
""""""
file_path = Path(os.path.join(os.getcwd(), "test-dummy-snap"))
file_format = FileFormat.ALT
with Snapshot(file_path, "w", format=file_format) as snap:
# Assign new header and block.
snap.header = dummy_snap_header
snap["POS"] = dummy_snap_block
print(snap["POS"])
snap.flush()
def test_inspect(snapshot_path):
""""""
with Snapshot(snapshot_path) as snap:
block_specs = snap.inspect()
print(block_specs)
def test_read(snapshot_path):
""""""
with Snapshot(snapshot_path) as snap:
print(snap)
def test_read_created(dummy_snap_header):
""""""
file_path = Path(os.path.join(os.getcwd(), "test-dummy-snap"))
with Snapshot(file_path) as snap:
assert snap.header == dummy_snap_header
print(snap["POS"])