forked from Archives/langchain
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
16 lines
541 B
Python
16 lines
541 B
Python
from langchain.chains.graph_qa.cypher import extract_cypher
|
|
|
|
|
|
def test_no_backticks() -> None:
|
|
"""Test if there are no backticks, so the original text should be returned."""
|
|
query = "MATCH (n) RETURN n"
|
|
output = extract_cypher(query)
|
|
assert output == query
|
|
|
|
|
|
def test_backticks() -> None:
|
|
"""Test if there are backticks. Query from within backticks should be returned."""
|
|
query = "You can use the following query: ```MATCH (n) RETURN n```"
|
|
output = extract_cypher(query)
|
|
assert output == "MATCH (n) RETURN n"
|