Reactivate suggest test (#1394)

* Reactivate suggest test * Fix suggest tests for new relations index
nuclia · Jan 4, 2024 · 5d7b87a · 5d7b87a
1 parent 8ad4b5e
commit 5d7b87a
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/nucliadb/nucliadb/tests/integration/test_suggest.py b/nucliadb/nucliadb/tests/integration/test_suggest.py
@@ -322,8 +322,8 @@ async def test_suggestion_on_link_computed_titles_sc6088(
     assert suggested["text"] == extracted_title
 
 
-@pytest.mark.skip(reason="Bindings not released")
 @pytest.mark.asyncio
+@pytest.mark.parametrize("knowledgebox", ("EXPERIMENTAL", "STABLE"), indirect=True)
 async def test_suggest_features(
     nucliadb_grpc: WriterStub,
     nucliadb_reader: AsyncClient,
@@ -349,8 +349,9 @@ def assert_expected_paragraphs(response):
         }
 
     def assert_expected_entities(response):
-        assert response["entities"]["total"] == 1
-        assert set(response["entities"]["entities"]) == {"Anna"}
+        expected = {"Anastasia", "Anna", "Anthony"}
+        assert response["entities"]["total"] == len(expected)
+        assert set(response["entities"]["entities"]) == expected
 
     resp = await nucliadb_reader.get(
         f"/kb/{knowledgebox}/suggest",