Make the update() method returns an Index object instead of a dict

Author: curquiza

meilisearch/index.py

@@ -55,16 +55,16 @@ def update(self, **body):
 
         Returns
         -------
-        index: dict
-            Dictionary containing index information.
+        index : Index
+            An Index instance of the newly updated index.
         """
         payload = {}
         primary_key = body.get('primaryKey', None)
         if primary_key is not None:
             payload['primaryKey'] = primary_key
         response = self.http.put('{}/{}'.format(self.config.paths.index, self.uid), payload)
         self.primary_key = response['primaryKey']
-        return response
+        return self
 
     def fetch_info(self):
         """Fetch the info of the index.

meilisearch/tests/index/test_index.py

@@ -139,7 +139,8 @@ def test_update_index(self):
         """Tests updating an index"""
         index = self.client.index(uid=self.index_uid)
         response = index.update(primaryKey='objectID')
-        assert isinstance(response, object)
+        assert response.uid == index.uid
+        assert response.primary_key == index.primary_key
         assert index.primary_key == 'objectID'
         assert index.get_primary_key() == 'objectID'