Skip to content

Commit

Permalink
MODIFIED: test cases file
Browse files Browse the repository at this point in the history
  • Loading branch information
AquibPy committed Mar 13, 2024
1 parent b5e9238 commit 4276a74
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions test_main.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@


def test_invoice_extractor():
image_file = open("data\invoice.png", "rb")
image_file = open("data/invoice.png", "rb")
files = {"image_file": image_file}
data = {"prompt": "To whom this invoice belongs"}

Expand Down Expand Up @@ -37,23 +37,23 @@ def test_youtube_video_transcribe_summarizer():
assert "response" in response.json()

def test_nutritionist_expert():
image_file = open(r"data\burger.jpg", "rb")
image_file = open("data/burger.jpg", "rb")
data = {"height": "165", "weight": "70"}

response = client.post("/nutritionist_expert", files={"image_file": image_file}, data=data)
assert response.status_code == 200
assert "response" in response.json()

def test_talk2PDF():
pdf_file = open(r"data\yolo.pdf", "rb")
pdf_file = open("data/yolo.pdf", "rb")
data = {"prompt": "Summary in 200 words"}

response = client.post("/talk2PDF", files={"pdf": pdf_file}, data=data)
assert response.status_code == 200
assert "response" in response.json()

def test_questions_generator():
pdf_file = open(r"data\yolo.pdf", "rb")
pdf_file = open("data/yolo.pdf", "rb")

# Make a request to the questions_generator endpoint
response = client.post("/questions_generator", files={"pdf": pdf_file})
Expand Down

0 comments on commit 4276a74

Please sign in to comment.