Create test_case_generator.py

tools/test_case_generator.py

@@ -0,0 +1,40 @@
+from typing import List, Dict
+from langchain.tools import tool
+
+@tool
+def suggest_test_cases(topic: str) -> List[Dict[str, str]]:
+    """
+    Suggest simple QA test cases for a given topic.
+
+    Args:
+        topic: The feature, function, or topic to generate test cases for.
+
+    Returns:
+        A list of test cases, each with input and expected output.
+    """
+    topic_lower = topic.lower()
+    test_cases = []
+
+    if "login" in topic_lower:
+        test_cases = [
+            {"test_case": "Valid login", "input": "Correct username/password", "expected_output": "User is logged in successfully"},
+            {"test_case": "Invalid password", "input": "Wrong password", "expected_output": "Error message displayed"},
+            {"test_case": "Empty username", "input": "Username left blank", "expected_output": "Error message displayed"},
+            {"test_case": "Empty password", "input": "Password left blank", "expected_output": "Error message displayed"},
+        ]
+    elif "calculator" in topic_lower:
+        test_cases = [
+            {"test_case": "Addition", "input": "2 + 3", "expected_output": "5"},
+            {"test_case": "Subtraction", "input": "5 - 2", "expected_output": "3"},
+            {"test_case": "Multiplication", "input": "4 * 5", "expected_output": "20"},
+            {"test_case": "Divide by zero", "input": "5 / 0", "expected_output": "Error handled gracefully"},
+        ]
+    else:
+        # Generic template
+        test_cases = [
+            {"test_case": f"Test case 1 for {topic}", "input": "Sample input", "expected_output": "Expected behavior"},
+            {"test_case": f"Test case 2 for {topic}", "input": "Another input", "expected_output": "Expected behavior"},
+            {"test_case": f"Test case 3 for {topic}", "input": "Additional input", "expected_output": "Expected behavior"},
+        ]
+
+    return test_cases