def test_docnetdb_vertices(tmp_path):
"""Test if the DocNetDB vertices() returns all inserted vertices."""
db = DocNetDB(tmp_path / "db.db")
v1, v2, v3 = Vertex(), Vertex(), Vertex()
for vertex in v1, v2, v3:
db.insert(vertex)
assert list(db.vertices()) == [v1, v2, v3]
def test_docnetdb_load_vertices(tmp_path):
"""Test if the DocNetDB load restores all the vertices in the object."""
path = tmp_path / "db.db"
db1 = DocNetDB(path)
music_names = ["Prologue", "First Steps", "Resurrections"]
for name in music_names:
db1.insert(Vertex({"name": name}))
db1.save()
db2 = DocNetDB(path)
assert len(db1) == len(db2)
for vertex, target_name in zip(db2.vertices(), music_names):
assert vertex["name"] == target_name
