From 3a7383425f473f88fba571020cecde8371f4c8b2 Mon Sep 17 00:00:00 2001 From: Siddhesh Agarwal Date: Fri, 1 Nov 2024 11:09:54 +0530 Subject: [PATCH] sorted imports --- tests/conftest.py | 4 ++-- tests/test_generate_data.py | 8 ++++---- tests/test_generate_text.py | 3 +-- 3 files changed, 7 insertions(+), 8 deletions(-) diff --git a/tests/conftest.py b/tests/conftest.py index 663d5b5..849d89d 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -1,8 +1,8 @@ -import pytest - import os import sys +import pytest + # Add the project root to the Python path. sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) diff --git a/tests/test_generate_data.py b/tests/test_generate_data.py index bea35c8..3d8a6e3 100644 --- a/tests/test_generate_data.py +++ b/tests/test_generate_data.py @@ -1,8 +1,8 @@ -import pytest - -from simplemind.providers import Anthropic, Gemini, OpenAI, Groq, Ollama from pydantic import BaseModel +import pytest +from simplemind.providers import Anthropic, Gemini, Groq, Ollama, OpenAI + class ResponseModel(BaseModel): result: int @@ -25,4 +25,4 @@ def test_generate_data(provider_cls): data = provider.structured_response(prompt=prompt, response_model=ResponseModel) assert isinstance(data, ResponseModel) - assert type(data.result) == int + assert isinstance(data.result, int) diff --git a/tests/test_generate_text.py b/tests/test_generate_text.py index 55a9a13..80f062d 100644 --- a/tests/test_generate_text.py +++ b/tests/test_generate_text.py @@ -1,6 +1,5 @@ import pytest - -from simplemind.providers import Anthropic, Gemini, OpenAI, Groq, Ollama +from simplemind.providers import Anthropic, Gemini, Groq, Ollama, OpenAI @pytest.mark.parametrize(