tursom/memgraph
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Improve tests

Browse Source
This commit is contained in:
Ante Pušić 2024-01-18 01:51:28 +01:00
parent 2df3b09019
commit c60b66fde7
2 changed files with 11 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/e2e/text_search/common.py
View File

@ -61,7 +61,7 @@ def memgraph_with_text_indexed_data(**kwargs) -> Memgraph:
@pytest.fixture
def memgraph_with_text_indexed_data(**kwargs) -> Memgraph:
def memgraph_with_mixed_data(**kwargs) -> Memgraph:
memgraph = Memgraph()
memgraph.execute("CREATE (:Document:Revision {title: 'Rules2024', version: 1});")

13
tests/e2e/text_search/test_text_search.py
View File

@ -23,12 +23,17 @@ GET_RULES_2024_DOCUMENT = """CALL text_search.search('complianceDocuments', 'dat
def test_create_index(memgraph):
memgraph.execute("CREATE TEXT INDEX exampleIndex ON :Document;")
assert True
index_info = memgraph.execute_and_fetch("SHOW INDEX INFO")
assert list(index_info) == [{"index type": "text", "label": "exampleIndex", "property": None, "count": None}]
def test_drop_index(memgraph):
memgraph.execute("DROP TEXT INDEX exampleIndex;")
assert True
index_info = memgraph.execute_and_fetch("SHOW INDEX INFO")
assert list(index_info) == []
def test_text_search_given_property(memgraph_with_text_indexed_data):
@ -38,6 +43,7 @@ def test_text_search_given_property(memgraph_with_text_indexed_data):
def test_text_search_all_properties(memgraph_with_text_indexed_data):
# json issue
result = list(
memgraph_with_text_indexed_data.execute_and_fetch(
"""CALL text_search.search('complianceDocuments', 'Rules2024') YIELD node
@ -46,7 +52,7 @@ def test_text_search_all_properties(memgraph_with_text_indexed_data):
)
)
# assert len(result) == 2 and result == [{"title": "Rules2024", "version": 1}, {"title": "Rules2024", "version": 2}]
assert len(result) == 2 and result == [{"title": "Rules2024", "version": 1}, {"title": "Rules2024", "version": 2}]
def test_create_indexed_node(memgraph_with_text_indexed_data):
@ -70,6 +76,7 @@ def test_delete_indexed_node(memgraph_with_text_indexed_data):
def test_add_indexed_label(memgraph_with_mixed_data):
# flaky search results
memgraph_with_mixed_data.execute("MATCH (n:Revision) SET n:Document;")
result = list(memgraph_with_mixed_data.execute_and_fetch(GET_RULES_2024_DOCUMENT))