LLM/DocsGPT
1
0
Fork 0
mirror of https://github.com/arc53/DocsGPT.git synced 2025-11-29 16:43:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: tests

Browse Source
This commit is contained in:
Alex
2024-10-15 11:52:33 +01:00
parent 39e94d4a5e
commit cca6297a64
3 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/llm/test_sagemaker.py
View File

@@ -52,7 +52,7 @@ class TestSagemakerAPILLM(unittest.TestCase):
self.response['Body'].read.return_value.decode.return_value = json.dumps(self.result)
def test_gen(self):
with patch('application.cache.make_redis') as mock_make_redis:
with patch('application.cache.get_redis_instance') as mock_make_redis:
mock_redis_instance = mock_make_redis.return_value
mock_redis_instance.get.return_value = None
@@ -70,7 +70,7 @@ class TestSagemakerAPILLM(unittest.TestCase):
mock_redis_instance.set.assert_called_once()
def test_gen_stream(self):
with patch('application.cache.make_redis') as mock_make_redis:
with patch('application.cache.get_redis_instance') as mock_make_redis:
mock_redis_instance = mock_make_redis.return_value
mock_redis_instance.get.return_value = None