Add fullset to main test

tests/test_main.py

@@ -21,6 +21,23 @@ def test_with_subset():
     question = "How should I train for anerobic capacity?"
     print("Question: ", question)
     
+    answer = run_query(question,
+                   db_path=db_path,
+                   num_rel_segments=10,
+                   llm_model="gpt-3.5-turbo-0125",
+                   llm_temp=0.1
+                   )
+    
+    print("Answer: ", answer)    
+    
+    
+@timeit
+def test_with_fullset():
+    db_path = "data/videos.db"
+    
+    question = "How should I train for anerobic capacity?"
+    print("Question: ", question)
+    
     answer = run_query(question,
                    db_path=db_path,
                    num_rel_segments=10,
@@ -32,11 +49,13 @@ def test_with_subset():
 
 
 if __name__ == '__main__':
-    choice = input("Enter 1 for test_with_subset, 2 for test_with_llm_video: ")
+    choice = input("Test with:\n1. fullset\n2. subset\n3. LLM video\nEnter option number: ")
     
     if choice == "1":
-        test_with_subset()
+        test_with_fullset()
     elif choice == "2":
+        test_with_subset()
+    elif choice == "3":
         test_with_llm_video()
     else:
         print("Invalid choice")