Update test_gpt4.py

anupammaurya6767 · Feb 29, 2024 · 7d61eca · 7d61eca
1 parent b29ce1d
commit 7d61eca
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/test_gpt4.py b/test_gpt4.py
@@ -12,9 +12,15 @@ def test_login(self):
     def test_ask_question(self):
         question = 'Test question'
         self.ap.login()
-        self.ap.ask_question(question,10)
+        self.ap.ask_question(question,20)
         response = self.ap.get_response()
         pass
+
+    def test_design(self):
+        question = 'A cow'
+        self.ap.login()
+        self.ap.design(question)
+        pass
 
     def tearDown(self):
         self.ap.close()