1+ import logging
12import os
3+ from unittest .mock import MagicMock , patch
4+
25import pytest
3- import logging
4- from unittest .mock import patch , MagicMock
6+
57from backend .generate_quiz import QuizGenerator
68
79"""
@@ -50,10 +52,8 @@ def test_environment_variable_not_set(self, monkeypatch):
5052 monkeypatch .delenv ("GEMINI_API_KEY" , raising = False )
5153 monkeypatch .delenv ("DEEPSEEK_API_KEY" , raising = False )
5254 monkeypatch .delenv ("AZURE_AI_API_KEY" , raising = False )
53-
54- with pytest .raises (
55- ValueError , match = "No API keys found"
56- ):
55+
56+ with pytest .raises (ValueError , match = "No API keys found" ):
5757 QuizGenerator ()
5858
5959 def test_create_role (self , quiz_generator ):
@@ -109,7 +109,7 @@ def test_model(self, model):
109109 """Test each supported model individually."""
110110 # Check if required API keys are available
111111 provider = model .split ("/" )[0 ]
112-
112+
113113 if provider == "openai" and not os .getenv ("OPENAI_API_KEY" ):
114114 pytest .skip ("OPENAI_API_KEY not set" )
115115 elif provider == "gemini" and not os .getenv ("GEMINI_API_KEY" ):
@@ -120,12 +120,12 @@ def test_model(self, model):
120120 not os .getenv ("AZURE_AI_API_KEY" ) or not os .getenv ("AZURE_AI_API_BASE" )
121121 ):
122122 pytest .skip ("Azure AI credentials not set" )
123-
123+
124124 # Test the model
125125 quiz_gen = QuizGenerator (model = model )
126126 gen = quiz_gen .generate_quiz ("Math" , "Easy" , 1 )
127127 results = list (gen )
128-
128+
129129 assert len (results ) > 0
130130 for r in results :
131131 assert r .startswith ("data: " )
0 commit comments