Fix: Make api test marked, and give keys to generators

SverreNystad · Sep 19, 2023 · eaa7597 · eaa7597
1 parent 6b1c8d3
commit eaa7597
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/text_generation/test_text_generator.py b/tests/text_generation/test_text_generator.py
@@ -1,17 +1,18 @@
 import pytest
 from src.text_generation.text_generator import get_default_text_generator, TextGenerator, LLM, Chatbot
 
+@pytest.mark.apitest
 def test_get_default_text_generator():
     generator: TextGenerator = get_default_text_generator()
     assert generator is not None, "The generator was None"
     assert isinstance(generator, TextGenerator), "The generator was not an TextGenerator"
 
 def test_LLM():
-    generator: TextGenerator = LLM()
+    generator: TextGenerator = LLM("DUMMY_KEY")
     assert isinstance(generator, TextGenerator), "The generator was not an TextGenerator"
 
 def test_Chatbot():
-    generator: TextGenerator = Chatbot()
+    generator: TextGenerator = Chatbot("DUMMY_KEY")
     assert isinstance(generator, TextGenerator), "The generator was not an TextGenerator"