Added a test for invalid inputs for parse_raw_prompts

Signed-off-by: Kayvan Mivehnejad <[email protected]>

Author: mivehk

tests/test_inputs.py

@@ -33,6 +33,16 @@
     slice(None, None, -2),
 ]
 
+@pytest.mark.parametrize(
+    "invalid_input",
+    [
+        ["foo", 1], # mixed of string and token
+        [["foo"], ["bar"]] # list of list of strings
+    ]
+)
+def test_invalid_input_raise_type_error(invalid_input):
+    with pytest.raises(TypeError):
+        parse_raw_prompts(invalid_input)
 
 # Test that a nested mixed-type list of lists raises a TypeError.
 @pytest.mark.parametrize("invalid_input", [[[1, 2], ["foo", "bar"]]])