From 71aeda669eb86a6f902f64b064aeec722ee7dfc5 Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Fri, 3 Nov 2023 09:37:23 -0300
Subject: [PATCH] =?UTF-8?q?=F0=9F=94=A8=20refactor(test=5Fchains=5Ftemplat?=
 =?UTF-8?q?e.py):=20remove=20unused=20test=20functions=20to=20improve=20co?=
 =?UTF-8?q?de=20readability=20and=20maintainability?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 tests/test_chains_template.py | 164 ----------------------------------
 1 file changed, 164 deletions(-)

diff --git a/tests/test_chains_template.py b/tests/test_chains_template.py
index ecbb9445c..dd6c2f058 100644
--- a/tests/test_chains_template.py
+++ b/tests/test_chains_template.py
@@ -9,170 +9,6 @@ from fastapi.testclient import TestClient
 #     assert set(chains.keys()) == set(settings.chains)
 
 
-# Test the ConversationChain object
-def test_conversation_chain(client: TestClient, logged_in_headers):
-    response = client.get("api/v1/all", headers=logged_in_headers)
-    assert response.status_code == 200
-    json_response = response.json()
-    chains = json_response["chains"]
-
-    chain = chains["ConversationChain"]
-    # Test the base classes, template, memory, verbose, llm, input_key, output_key, and _type objects
-    assert set(chain["base_classes"]) == {
-        "ConversationChain",
-        "LLMChain",
-        "Chain",
-        "Callable",
-    }
-
-    template = chain["template"]
-    assert template["memory"] == {
-        "required": False,
-        "dynamic": False,
-        "placeholder": "",
-        "show": True,
-        "multiline": False,
-        "password": False,
-        "name": "memory",
-        "type": "BaseMemory",
-        "list": False,
-        "advanced": False,
-        "info": "",
-    }
-    assert template["verbose"] == {
-        "required": False,
-        "dynamic": False,
-        "placeholder": "",
-        "show": False,
-        "multiline": False,
-        "password": False,
-        "name": "verbose",
-        "type": "bool",
-        "list": False,
-        "advanced": True,
-        "info": "",
-    }
-    assert template["llm"] == {
-        "required": True,
-        "dynamic": False,
-        "placeholder": "",
-        "show": True,
-        "multiline": False,
-        "password": False,
-        "name": "llm",
-        "type": "BaseLanguageModel",
-        "list": False,
-        "advanced": False,
-        "info": "",
-    }
-    assert template["input_key"] == {
-        "required": True,
-        "dynamic": False,
-        "placeholder": "",
-        "show": True,
-        "multiline": False,
-        "value": "input",
-        "password": False,
-        "name": "input_key",
-        "type": "str",
-        "list": False,
-        "advanced": True,
-        "info": "",
-    }
-    assert template["output_key"] == {
-        "required": True,
-        "dynamic": False,
-        "placeholder": "",
-        "show": True,
-        "multiline": False,
-        "value": "response",
-        "password": False,
-        "name": "output_key",
-        "type": "str",
-        "list": False,
-        "advanced": True,
-        "info": "",
-    }
-    assert template["_type"] == "ConversationChain"
-
-    # Test the description object
-    assert (
-        chain["description"]
-        == "Chain to have a conversation and load context from memory."
-    )
-
-
-def test_llm_chain(client: TestClient, logged_in_headers):
-    response = client.get("api/v1/all", headers=logged_in_headers)
-    assert response.status_code == 200
-    json_response = response.json()
-    chains = json_response["chains"]
-    chain = chains["LLMChain"]
-
-    # Test the base classes, template, memory, verbose, llm, input_key, output_key, and _type objects
-    assert set(chain["base_classes"]) == {
-        "Callable",
-        "LLMChain",
-        "Chain",
-    }
-
-    template = chain["template"]
-    assert template["memory"] == {
-        "required": False,
-        "dynamic": False,
-        "placeholder": "",
-        "show": True,
-        "multiline": False,
-        "password": False,
-        "name": "memory",
-        "type": "BaseMemory",
-        "list": False,
-        "advanced": False,
-        "info": "",
-    }
-    assert template["verbose"] == {
-        "required": False,
-        "dynamic": False,
-        "placeholder": "",
-        "show": False,
-        "multiline": False,
-        "value": False,
-        "password": False,
-        "name": "verbose",
-        "type": "bool",
-        "list": False,
-        "advanced": True,
-        "info": "",
-    }
-    assert template["llm"] == {
-        "required": True,
-        "dynamic": False,
-        "placeholder": "",
-        "show": True,
-        "multiline": False,
-        "password": False,
-        "name": "llm",
-        "type": "BaseLanguageModel",
-        "list": False,
-        "advanced": False,
-        "info": "",
-    }
-    assert template["output_key"] == {
-        "required": True,
-        "dynamic": False,
-        "placeholder": "",
-        "show": True,
-        "multiline": False,
-        "value": "text",
-        "password": False,
-        "name": "output_key",
-        "type": "str",
-        "list": False,
-        "advanced": True,
-        "info": "",
-    }
-
-
 def test_llm_checker_chain(client: TestClient, logged_in_headers):
     response = client.get("api/v1/all", headers=logged_in_headers)
     assert response.status_code == 200