From ccee46b8c6815c5de9aca6e3a805a917e272e269 Mon Sep 17 00:00:00 2001 From: Oguz Vuruskaner Date: Sat, 4 May 2024 18:05:58 +0300 Subject: [PATCH] test(body test): implement --- tests/test_embeddings.py | 5 +++-- tests/test_text_generation.py | 3 +++ tests/test_text_to_image.py | 3 +++ 3 files changed, 9 insertions(+), 2 deletions(-) diff --git a/tests/test_embeddings.py b/tests/test_embeddings.py index 56a543a..67cdb14 100644 --- a/tests/test_embeddings.py +++ b/tests/test_embeddings.py @@ -1,3 +1,4 @@ +import json import unittest from unittest.mock import patch @@ -24,9 +25,9 @@ def test_generate(self, mock_post): called_args, called_kwargs = mock_post.call_args url = called_args[0] header = called_kwargs["headers"] - body = called_kwargs["data"] + data = called_kwargs["data"] self.assertEqual(url, f"https://api.deepinfra.com/v1/inference/{model_name}") - self.assertEqual(body, '{"text": "Hello, World!"}') + self.assertEqual(data, json.dumps(body)) self.assertEqual(response.embeddings, [1, 2, 3]) self.assertEqual(header["Authorization"], f"Bearer {api_key}") diff --git a/tests/test_text_generation.py b/tests/test_text_generation.py index 56501b4..233c572 100644 --- a/tests/test_text_generation.py +++ b/tests/test_text_generation.py @@ -1,3 +1,4 @@ +import json import unittest from unittest.mock import patch @@ -25,8 +26,10 @@ def test_generate(self, mock_post): called_args, called_kwargs = mock_post.call_args url = called_args[0] + data = called_kwargs["data"] header = called_kwargs["headers"] self.assertEqual(url, f"https://api.deepinfra.com/v1/inference/{model_name}") self.assertEqual(response.results, []) self.assertEqual(header["Authorization"], f"Bearer {api_key}") + self.assertEqual(data, json.dumps(body)) diff --git a/tests/test_text_to_image.py b/tests/test_text_to_image.py index 5c75fd8..6088534 100644 --- a/tests/test_text_to_image.py +++ b/tests/test_text_to_image.py @@ -1,3 +1,4 @@ +import json import unittest from unittest.mock import patch @@ -30,7 +31,9 @@ def test_generate(self, mock_post): called_args, called_kwargs = mock_post.call_args url = called_args[0] header = called_kwargs["headers"] + data = called_kwargs["data"] self.assertEqual(url, f"https://api.deepinfra.com/v1/inference/{model_name}") self.assertEqual(response.images, images) self.assertEqual(header["Authorization"], f"Bearer {api_key}") + self.assertEqual(data, json.dumps(body))