async def METHOD_NAME(self, client, doc_count):
    docs = [{"hotelId": "5", "rating": 1}, {"hotelId": "6", "rating": 2}]
    results = await client.merge_documents(docs)
    assert len(results) == len(docs)
    assert set(x.status_code for x in results) == {200}
    # There can be some lag before a document is searchable
    if self.is_live:
        time.sleep(TIME_TO_SLEEP)
    assert await client.get_document_count() == doc_count
    result = await client.get_document(key="5")
    assert result["rating"] == 1
    result = await client.get_document(key="6")
    assert result["rating"] == 2
    return doc_count