From 8874eafa7533af80f11041af3d6da0c2abcde699 Mon Sep 17 00:00:00 2001
From: NolanB <nboukachab@teklia.com>
Date: Mon, 5 Sep 2022 15:04:17 +0200
Subject: [PATCH] Fix lintin
---
arkindex_worker/worker/entity.py | 3 --
.../test_classifications.py | 24 +++++++++++----
tests/test_elements_worker/test_elements.py | 30 +++++++++++++++----
3 files changed, 42 insertions(+), 15 deletions(-)
diff --git a/arkindex_worker/worker/entity.py b/arkindex_worker/worker/entity.py
index 173b6c34..5e2b6cf6 100644
--- a/arkindex_worker/worker/entity.py
+++ b/arkindex_worker/worker/entity.py
@@ -40,9 +40,6 @@ class EntityMixin(object):
:type element: Element or CachedElement
:param name str: Name of the entity.
:param type EntityType: Type of the entity.
- :param corpus: UUID of the corpus to create an entity on, or None to use the
- value of the ``ARKINDEX_CORPUS_ID`` environment variable.
- :type corpus: str or None
"""
assert element and isinstance(
element, (Element, CachedElement)
diff --git a/tests/test_elements_worker/test_classifications.py b/tests/test_elements_worker/test_classifications.py
index 047f9ae6..e6831b32 100644
--- a/tests/test_elements_worker/test_classifications.py
+++ b/tests/test_elements_worker/test_classifications.py
@@ -36,7 +36,10 @@ def test_get_ml_class_id_load_classes(responses, mock_elements_worker):
assert [
(call.request.method, call.request.url) for call in responses.calls
] == BASE_API_CALLS + [
- ("GET", f"http://testserver/api/v1/corpus/{mock_elements_worker.corpus_id}/classes/"),
+ (
+ "GET",
+ f"http://testserver/api/v1/corpus/{mock_elements_worker.corpus_id}/classes/",
+ ),
]
assert mock_elements_worker.classes == {
"12341234-1234-1234-1234-123412341234": {"good": "0000"}
@@ -46,9 +49,9 @@ def test_get_ml_class_id_load_classes(responses, mock_elements_worker):
def test_get_ml_class_id_inexistant_class(mock_elements_worker, responses):
# A missing class is now created automatically
- mock_elements_worker.corpus_id = "12341234-1234-1234-1234-123412341234"
+ mock_elements_worker.corpus_id = "12341234-1234-1234-1234-123412341234"
mock_elements_worker.classes = {
- "12341234-1234-1234-1234-123412341234": {"good": "0000"}
+ "12341234-1234-1234-1234-123412341234": {"good": "0000"}
}
responses.add(
@@ -145,9 +148,18 @@ def test_get_ml_class_reload(responses, mock_elements_worker):
assert [
(call.request.method, call.request.url) for call in responses.calls
] == BASE_API_CALLS + [
- ("GET", f"http://testserver/api/v1/corpus/{mock_elements_worker.corpus_id}/classes/"),
- ("POST", f"http://testserver/api/v1/corpus/{mock_elements_worker.corpus_id}/classes/"),
- ("GET", f"http://testserver/api/v1/corpus/{mock_elements_worker.corpus_id}/classes/"),
+ (
+ "GET",
+ f"http://testserver/api/v1/corpus/{mock_elements_worker.corpus_id}/classes/",
+ ),
+ (
+ "POST",
+ f"http://testserver/api/v1/corpus/{mock_elements_worker.corpus_id}/classes/",
+ ),
+ (
+ "GET",
+ f"http://testserver/api/v1/corpus/{mock_elements_worker.corpus_id}/classes/",
+ ),
]
diff --git a/tests/test_elements_worker/test_elements.py b/tests/test_elements_worker/test_elements.py
index 6ad91b51..97ae3a56 100644
--- a/tests/test_elements_worker/test_elements.py
+++ b/tests/test_elements_worker/test_elements.py
@@ -289,11 +289,26 @@ def test_load_corpus_classes_api_error(responses, mock_elements_worker):
(call.request.method, call.request.url) for call in responses.calls
] == BASE_API_CALLS + [
# We do 5 retries
- ("GET", f"http://testserver/api/v1/corpus/{mock_elements_worker.corpus_id}/classes/"),
- ("GET", f"http://testserver/api/v1/corpus/{mock_elements_worker.corpus_id}/classes/"),
- ("GET", f"http://testserver/api/v1/corpus/{mock_elements_worker.corpus_id}/classes/"),
- ("GET", f"http://testserver/api/v1/corpus/{mock_elements_worker.corpus_id}/classes/"),
- ("GET", f"http://testserver/api/v1/corpus/{mock_elements_worker.corpus_id}/classes/"),
+ (
+ "GET",
+ f"http://testserver/api/v1/corpus/{mock_elements_worker.corpus_id}/classes/",
+ ),
+ (
+ "GET",
+ f"http://testserver/api/v1/corpus/{mock_elements_worker.corpus_id}/classes/",
+ ),
+ (
+ "GET",
+ f"http://testserver/api/v1/corpus/{mock_elements_worker.corpus_id}/classes/",
+ ),
+ (
+ "GET",
+ f"http://testserver/api/v1/corpus/{mock_elements_worker.corpus_id}/classes/",
+ ),
+ (
+ "GET",
+ f"http://testserver/api/v1/corpus/{mock_elements_worker.corpus_id}/classes/",
+ ),
]
assert not mock_elements_worker.classes
@@ -331,7 +346,10 @@ def test_load_corpus_classes(responses, mock_elements_worker):
assert [
(call.request.method, call.request.url) for call in responses.calls
] == BASE_API_CALLS + [
- ("GET", f"http://testserver/api/v1/corpus/{mock_elements_worker.corpus_id}/classes/"),
+ (
+ "GET",
+ f"http://testserver/api/v1/corpus/{mock_elements_worker.corpus_id}/classes/",
+ ),
]
assert mock_elements_worker.classes == {
"12341234-1234-1234-1234-123412341234": {
--
GitLab