Fix update document test after update to v0.29

Author: brunoocasali

tests/index/test_index_document_meilisearch.py

@@ -1,6 +1,7 @@
 # pylint: disable=invalid-name
 
 from math import ceil
+from time import sleep
 from meilisearch.client import Client
 from meilisearch.models.document import Document
 from meilisearch.models.task import TaskInfo
@@ -91,17 +92,23 @@ def test_update_documents(index_with_documents, small_movies):
     """Tests updating a single document and a set of documents."""
     index = index_with_documents()
     response = index.get_documents()
-    response.results[0].title = 'Some title'
-    update = index.update_documents([dict(response.results[0])])
+    doc = response.results[0]
+    doc.title = 'Some title'
+
+    update = index.update_documents([dict(doc)])
+
     assert isinstance(update, TaskInfo)
     assert update.task_uid != None
     index.wait_for_task(update.task_uid)
-    response = index.get_documents()
-    assert response.results[0].title == 'Some title'
+
+    response = index.get_document(doc.id)
+    assert response.title == 'Some title'
+
     update = index.update_documents(small_movies)
     index.wait_for_task(update.task_uid)
-    response = index.get_documents()
-    assert response.results[0].title != 'Some title'
+
+    response = index.get_document(doc.id)
+    assert response.title != 'Some title'
 
 @pytest.mark.parametrize('batch_size', [2, 3, 1000])
 @pytest.mark.parametrize(