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(