Adding a test file

test_app.py

@@ -0,0 +1,31 @@
+# This basis of this code was generated by ChatGPT. The following prompt was: I have a python script that creates a Chatbot. I want some example tests that
+# could be put in a test.py file that test if the Chatbot is functioning. 
+
+import os
+from fastapi.testclient import TestClient
+import pytest
+import gradio as gr
+from app import respond, update_system_message, cancel_inference, demo  # Import functions from app.py
+
+client = TestClient(demo)
+
+
+def test_update_system_message():
+    assert update_system_message("Elementary School") == (
+        "Your name is Wormington. You are a friendly Chatbot that can help answer questions from elementary school students. Please respond with the vocabulary that a seven-year-old can understand."
+    )
+    assert update_system_message("Middle School") == (
+        "Your name is Wormington. You are a friendly Chatbot that can help answer questions from middle school students. Please respond at a level that middle schoolers can understand."
+    )
+    assert update_system_message("High School") == (
+        "Your name is Wormington. You are a friendly Chatbot that can help answer questions from high school students. Please respond at a level that a high schooler can understand."
+    )
+    assert update_system_message("College") == (
+        "Your name is Wormington. You are a friendly Chatbot that can help answer questions from college students. Please respond using very advanced, college-level vocabulary."
+    )
+
+
+def test_cancel_inference():
+    global stop_inference
+    cancel_inference()
+    assert stop_inference is True