jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

fixes to tests

Browse source
This commit is contained in:
Gabriel Almeida 2023-04-29 07:38:06 -03:00
commit ffb2aad45d
5 changed files with 12 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

14
tests/test_chains_template.py
View file

@ -20,7 +20,12 @@ def test_conversation_chain(client: TestClient):
chain = chains["ConversationChain"]
# Test the base classes, template, memory, verbose, llm, input_key, output_key, and _type objects
assert set(chain["base_classes"]) == {"LLMChain", "ConversationChain", "Chain"}
assert set(chain["base_classes"]) == {
"function",
"LLMChain",
"ConversationChain",
"Chain",
}
template = chain["template"]
assert template["memory"] == {
"required": False,
@ -96,7 +101,7 @@ def test_llm_chain(client: TestClient):
chain = chains["LLMChain"]
# Test the base classes, template, memory, verbose, llm, input_key, output_key, and _type objects
assert set(chain["base_classes"]) == {"LLMChain", "Chain"}
assert set(chain["base_classes"]) == {"function", "LLMChain", "Chain"}
template = chain["template"]
assert template["memory"] == {
"required": False,
@ -154,7 +159,7 @@ def test_llm_checker_chain(client: TestClient):
chain = chains["LLMCheckerChain"]
# Test the base classes, template, memory, verbose, llm, input_key, output_key, and _type objects
assert set(chain["base_classes"]) == {"LLMCheckerChain", "Chain"}
assert set(chain["base_classes"]) == {"function", "LLMCheckerChain", "Chain"}
template = chain["template"]
assert template["memory"] == {
"required": False,
@ -231,7 +236,7 @@ def test_llm_math_chain(client: TestClient):
chain = chains["LLMMathChain"]
# Test the base classes, template, memory, verbose, llm, input_key, output_key, and _type objects
assert set(chain["base_classes"]) == {"LLMMathChain", "Chain"}
assert set(chain["base_classes"]) == {"function", "LLMMathChain", "Chain"}
template = chain["template"]
assert template["memory"] == {
"required": False,
@ -310,6 +315,7 @@ def test_series_character_chain(client: TestClient):
# Test the base classes, template, memory, verbose, llm, input_key, output_key, and _type objects
assert set(chain["base_classes"]) == {
"function",
"LLMChain",
"BaseCustomChain",
"Chain",