def test_docnetdb_search_return_type(tmp_path): """Test if the DocNetDB search returns a generator.""" db = DocNetDB(tmp_path / "db.db") def false_func(x): return False result = db.search(false_func) assert isinstance(result, Generator) is True
def test_docnetdb_search_keyerror_autocatch(tmp_path): """Test if the DocNetDB search catches KeyError automatically.""" db = DocNetDB(tmp_path / "db.db") v1 = Vertex({"special_element": "WOW !"}) v2 = Vertex() db.insert(v1) db.insert(v2) def find_special_element(vertex): return vertex["special_element"] == "WOW !" assert list(db.search(find_special_element)) == [v1]
def test_docnetdb_search(tmp_path): """Test if the DocNetDB search returns the right vertices.""" db = DocNetDB(tmp_path / "db.db") v1, v2, v3 = Vertex(), Vertex(), Vertex() for vertex in v1, v2, v3: db.insert(vertex) def find_the_second(vertex): return vertex.place == 2 result = db.search(find_the_second) assert list(result) == [v2]