use openai fake fixture, update exception

leondz · Sep 23, 2024 · ad501ef · ad501ef
1 parent 60f9e12
commit ad501ef
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/generators/test_openai.py b/tests/generators/test_openai.py
@@ -7,6 +7,7 @@
 
 import openai
 
+import garak.exception
 from garak.generators.openai import OpenAIGenerator
 
 
@@ -92,8 +93,9 @@ def test_openai_chat():
     print("test passed!")
 
 
+@pytest.mark.usefixtures("set_fake_env")
 def test_reasoning_switch():
-    with pytest.raises(ValueError):
+    with pytest.raises(garak.exception.BadGeneratorException):
         generator = OpenAIGenerator(
             name="o1-mini"
         )  # o1 models should use ReasoningGenerator