Fix models tests

matiasvallejosdev · Apr 9, 2024 · 8ff5641 · 8ff5641
1 parent 47b322c
commit 8ff5641
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/tests/test_memory.py b/tests/test_memory.py
@@ -96,7 +96,7 @@ def test_remove_memory_invalid_user_id(inmemory, user_id):
     user_id_invalid = 1234
     inmemory._initialize(user_id)
     memory = inmemory.remove(user_id_invalid)
-    assert memory == f"'User {user_id_invalid} does not exist.'"
+    assert memory == f"User {user_id_invalid} does not exist."
 
 
 def test_get_memory(inmemory, user_id, message):

diff --git a/tests/test_models.py b/tests/test_models.py
@@ -21,8 +21,8 @@ def test_model_instance(model):
     assert isinstance(model, OpenAIModel)
 
 
-def test_str_method(model):
-    assert str(model) == f"OpenAIModel: {model.model_engine}"
+def test_str_model_method(model):
+    assert str(model) == f"Model: OpenAI. Engine: {model.model_engine}. Temperature: {model.temperature}. Max Tokens: {model.max_tokens}."
 
 
 def test_chat_completion(model):