def test_get_document(self):
        """Also ensures that the _id is returned"""
        for index in self.indexes:
            with self.subTest(f"Index type: {index.type}. Index name: {index.name}"):
                self.add_documents(
                    config=self.config, add_docs_params=AddDocsParams(index_name=index.name, docs=[
                        {
                            "_id": "123",
                            "title1": "content 1",
                            "desc2": "content 2. blah blah blah"
                        }], auto_refresh=True, device="cpu",
                        tensor_fields=[] if isinstance(index, UnstructuredMarqoIndex) else None
                        )
                )
                self.assertEqual(
                    tensor_search.get_document_by_id(
                        config=self.config, index_name=index.name,
                        document_id="123"
                    ),
                    {
                        "_id": "123",
                        "title1": "content 1",
                        "desc2": "content 2. blah blah blah"
                    }
                )
Also ensures that the _id is returned
tests/integ_tests/tensor_search/integ_tests/test_get_document.py