From 7750565754944ed85bf217c49a7374be02676867 Mon Sep 17 00:00:00 2001
From: Valentin Rigal <rigal@teklia.com>
Date: Wed, 23 Feb 2022 16:18:37 +0100
Subject: [PATCH] Update tests

---
 tests/test_elements_worker/test_elements.py | 17 +++--------------
 1 file changed, 3 insertions(+), 14 deletions(-)

diff --git a/tests/test_elements_worker/test_elements.py b/tests/test_elements_worker/test_elements.py
index 02c04ca3..6e2f132b 100644
--- a/tests/test_elements_worker/test_elements.py
+++ b/tests/test_elements_worker/test_elements.py
@@ -976,17 +976,6 @@ def test_list_element_children_wrong_element(mock_elements_worker):
     )
 
 
-def test_list_element_children_wrong_best_class(mock_elements_worker):
-    elt = Element({"id": "12341234-1234-1234-1234-123412341234"})
-
-    with pytest.raises(AssertionError) as e:
-        mock_elements_worker.list_element_children(
-            element=elt,
-            best_class=1234,
-        )
-    assert str(e.value) == "best_class should be of type str or bool"
-
-
 def test_list_element_children_wrong_folder(mock_elements_worker):
     elt = Element({"id": "12341234-1234-1234-1234-123412341234"})
 
@@ -1031,15 +1020,15 @@ def test_list_element_children_wrong_type(mock_elements_worker):
     assert str(e.value) == "type should be of type str"
 
 
-def test_list_element_children_wrong_with_best_classes(mock_elements_worker):
+def test_list_element_children_wrong_with_classes(mock_elements_worker):
     elt = Element({"id": "12341234-1234-1234-1234-123412341234"})
 
     with pytest.raises(AssertionError) as e:
         mock_elements_worker.list_element_children(
             element=elt,
-            with_best_classes="not bool",
+            with_classes="not bool",
         )
-    assert str(e.value) == "with_best_classes should be of type bool"
+    assert str(e.value) == "with_classes should be of type bool"
 
 
 def test_list_element_children_wrong_with_corpus(mock_elements_worker):
-- 
GitLab