Skip to content

Commit d3106c9

Browse files
Update test_views.py
1 parent c148c7d commit d3106c9

File tree

1 file changed

+0
-4
lines changed

1 file changed

+0
-4
lines changed

‎src/tests/test_views.py‎

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -140,19 +140,16 @@ def test_unknown_model_return_error_on_chat_completion(client_with_settings_over
140140
assert response.status_code == 404
141141

142142

143-
# Test for get_memory_instance
144143
@patch('dialog.llm.agents.lcel.get_session')
145144
@patch('dialog.llm.agents.lcel.generate_memory_instance')
146145
def test_get_memory_instance(mock_generate_memory_instance, mock_get_session):
147146
mock_get_session.return_value = iter([MagicMock()])
148147
lcel.get_memory_instance('test_session')
149148
mock_generate_memory_instance.assert_called_once()
150149

151-
# Test for retriever
152150
def test_retriever():
153151
assert isinstance(lcel.retriever, DialogRetriever)
154152

155-
# Test for format_docs
156153
def test_format_docs():
157154
docs = [Document(page_content='doc1'), Document(page_content='doc2')]
158155
result = lcel.format_docs(docs)
@@ -185,7 +182,6 @@ def psql_memory():
185182

186183

187184
async def test_gen():
188-
#import gen from dialog/src/dialog/routers/openai and test it
189185
from src.dialog.routers.openai import ask_question_to_llm
190186

191187
_g = ask_question_to_llm(message="Hello")

0 commit comments

Comments
 (0)