feat: Updated src/test_main.py

sweepai · Oct 24, 2023 · 858bceb · 858bceb
1 parent 9fa2b98
commit 858bceb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/test_main.py b/src/test_main.py
@@ -29,5 +29,5 @@ def test_predict(mocker: MockerFixture):
  mock_image_open.return_value.convert.return_value = Image.new('L', (28, 28))
  client = TestClient(app)
  response = client.post("/predict/", files={"file": ("filename", io.BytesIO(), "image/png")})
- assert response.status_code == 200
- assert 'prediction' in response.json()
+ assert response.status_code == 200, "Expected status code to be 200"
+ assert 'prediction' in response.json(), "Expected 'prediction' in response"