Spaces:
Sleeping
Sleeping
Clement Vachet
commited on
Commit
·
7e709ed
1
Parent(s):
5c0db05
style: use double quotes
Browse files- tests/test_classifier.py +2 -2
tests/test_classifier.py
CHANGED
@@ -18,7 +18,7 @@ def requests():
|
|
18 |
|
19 |
@pytest.fixture
|
20 |
def response():
|
21 |
-
return [
|
22 |
|
23 |
def test_response(setup_pipeline, requests, response):
|
24 |
-
assert response == setup_pipeline.load_and_test(requests)[
|
|
|
18 |
|
19 |
@pytest.fixture
|
20 |
def response():
|
21 |
+
return ["virginica", "versicolor"]
|
22 |
|
23 |
def test_response(setup_pipeline, requests, response):
|
24 |
+
assert response == setup_pipeline.load_and_test(requests)["predictions"]
|