Python Как исправить модуль с помощью Assert_ Called ⇐ Python

[Anonymous]

Я пытаюсь создать имитируемый AsyncOpenSearch и все его методы, чтобы в моем модульном тесте, когда я использую AsyncOS.ping(), он гарантировал мне, что AsyncOpenSearch.native_ping был вызван или ожидается
Как мне исправить весь AsyncOpenSearch и Native_ping?
assert_await или Assert_Call завершается с ошибкой, но печать из Native_ping печатается.
import pytest
from unittest.mock import AsyncMock, patch
import asyncio

class OSConfig:
pass

# Different module: ops.my_module.AsyncOpenSearch
# class AsyncOpenSearch:
# async def native_ping(self):
# print("I have been called!")

class AsyncOS:
def __init__(self, config: OSConfig):
self.client: AsyncOpenSearch = AsyncOpenSearch()

async def ping(self):
await self.client.native_ping()

@pytest.mark.asyncio
async def test_ping_called():
with patch("ops.my_module.AsyncOpenSearch", new_callable=AsyncMock) as os_mock:
async_os = AsyncOS(config=OSConfig())
await async_os.ping()

#os_mock.native_ping.assert_called() # doesn't work
os_mock.native_ping.assert_awaited() # doesn't work


Подробнее здесь: https://stackoverflow.com/questions/790 ... ert-called