20 lines
615 B
Python
20 lines
615 B
Python
|
import os
|
||
|
|
|
||
|
|
from werkzeug.datastructures import FileStorage
|
||
|
|
|
||
|
|
from main import app
|
||
|
|
|
||
|
|
|
||
|
|
def test_upload_file():
|
||
|
|
with app.test_client() as test_client:
|
||
|
|
script_directory = os.path.dirname(os.path.abspath(__file__))
|
||
|
|
image_path = os.path.join(script_directory, "./img/tiger_cat/cat1.jpg")
|
||
|
|
image = FileStorage(
|
||
|
|
stream=open(image_path, "rb"),
|
||
|
|
filename="cat1.jpg",
|
||
|
|
content_type="image/jpeg",
|
||
|
|
)
|
||
|
|
|
||
|
|
response = test_client.post('/api/v1/detect-cat', data={'image': image}, content_type='multipart/form-data')
|
||
|
|
assert response.status_code == 200
|