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