From d4da9bbc86f54cb1e02882f8c3ee434a672f0baa Mon Sep 17 00:00:00 2001 From: Fahad Ali Khan Date: Tue, 19 Nov 2024 23:21:30 -0500 Subject: [PATCH] test: add unit test for unsupported file format in process_files --- tests/test_process_files_unsupported_file_format.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) create mode 100644 tests/test_process_files_unsupported_file_format.py diff --git a/tests/test_process_files_unsupported_file_format.py b/tests/test_process_files_unsupported_file_format.py new file mode 100644 index 0000000..09e900d --- /dev/null +++ b/tests/test_process_files_unsupported_file_format.py @@ -0,0 +1,12 @@ +import pytest +from utils import process_files + +def test_process_files_unsupported_file_format(): + files = ["unsupported_file_format.pdf"] # Unsupported format for process_files + api_key = "test_key" + model = "llama3-8b-8192" + mode = "detailed" + provider_name = "groq" + + with pytest.raises(Exception, match="Unsupported file format"): + process_files(files, api_key, model, mode, token_usage=True, provider_name=provider_name) \ No newline at end of file