From d0f4841a10fd543fcc3d12acdd9f9c2296eb213f Mon Sep 17 00:00:00 2001
From: 7000pctAUTO <lukems3823@gmail.com>
Date: Wed, 4 Feb 2026 12:33:47 +0000
Subject: [PATCH] Add test suite

---
 tests/test_providers.py | 57 +++++++++++++++++++++++++++++++++++++++++
 1 file changed, 57 insertions(+)
 create mode 100644 tests/test_providers.py

diff --git a/tests/test_providers.py b/tests/test_providers.py
new file mode 100644
index 0000000..6ae21a1
--- /dev/null
+++ b/tests/test_providers.py
@@ -0,0 +1,57 @@
+import pytest
+import sys
+from pathlib import Path
+from unittest.mock import Mock, AsyncMock, patch
+
+sys.path.insert(0, str(Path(__file__).parent.parent / "src"))
+
+from promptforge.providers.factory import ProviderFactory
+from promptforge.providers.openai import OpenAIProvider
+from promptforge.providers.anthropic import AnthropicProvider
+from promptforge.providers.ollama import OllamaProvider
+
+
+class TestProviderFactory:
+    def test_create_openai(self):
+        provider = ProviderFactory.create("openai", model="gpt-4")
+        assert isinstance(provider, OpenAIProvider)
+        assert provider.model == "gpt-4"
+
+    def test_create_anthropic(self):
+        provider = ProviderFactory.create("anthropic", model="claude-3")
+        assert isinstance(provider, AnthropicProvider)
+
+    def test_create_ollama(self):
+        provider = ProviderFactory.create("ollama", model="llama2")
+        assert isinstance(provider, OllamaProvider)
+
+    def test_create_unknown(self):
+        with pytest.raises(Exception):
+            ProviderFactory.create("unknown")
+
+
+class TestOpenAIProvider:
+    def test_provider_name(self):
+        provider = OpenAIProvider(api_key="test-key")
+        assert provider.name == "openai"
+
+    def test_provider_defaults(self):
+        provider = OpenAIProvider()
+        assert provider.model == "gpt-4"
+        assert provider.temperature == 0.7
+
+
+class TestAnthropicProvider:
+    def test_provider_name(self):
+        provider = AnthropicProvider(api_key="test-key")
+        assert provider.name == "anthropic"
+
+
+class TestOllamaProvider:
+    def test_provider_name(self):
+        provider = OllamaProvider()
+        assert provider.name == "ollama"
+
+    def test_provider_defaults(self):
+        provider = OllamaProvider(model="llama2")
+        assert provider.model == "llama2"
\ No newline at end of file