fix: do not mock get (#598)

2023-09-12 19:12:17 +02:00
parent 03146946fa
commit ac08638a63
1 changed files with 0 additions and 2 deletions
--- a/tests/embedchain/test_embedchain.py
+++ b/tests/embedchain/test_embedchain.py
@@ -10,13 +10,11 @@ class TestChromaDbHostsLoglevel(unittest.TestCase):
    os.environ["OPENAI_API_KEY"] = "test_key"

    @patch("chromadb.api.models.Collection.Collection.add")
-    @patch("chromadb.api.models.Collection.Collection.get")
    @patch("embedchain.embedchain.EmbedChain.retrieve_from_database")
    @patch("embedchain.llm.base.BaseLlm.get_answer_from_llm")
    @patch("embedchain.llm.base.BaseLlm.get_llm_model_answer")
    def test_whole_app(
        self,
-        _mock_get,
        _mock_add,
        _mock_ec_retrieve_from_database,
        _mock_get_answer_from_llm,