Skip to content

Commit

Permalink
Merge pull request #11 from Fahad-Ali-Khan-ca/main
Browse files Browse the repository at this point in the history
Add unit test for unsupported file format in `process_files`
  • Loading branch information
InderParmar authored Nov 20, 2024
2 parents 383ef17 + d4da9bb commit 18bebe1
Showing 1 changed file with 12 additions and 0 deletions.
12 changes: 12 additions & 0 deletions tests/test_process_files_unsupported_file_format.py
Original file line number Diff line number Diff line change
@@ -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)

0 comments on commit 18bebe1

Please sign in to comment.