fix for 3.7

sammchardy · Nov 19, 2024 · f23b030 · f23b030
1 parent 6968df2
commit f23b030
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tests/test_async_client_ws_futures_requests.py b/tests/test_async_client_ws_futures_requests.py
@@ -1,11 +1,15 @@
 import asyncio
-from unittest.mock import AsyncMock, patch
 import pytest
 
 from binance.exceptions import BinanceAPIException, BinanceWebsocketUnableToConnect
 from .test_get_order_book import assert_ob
 from .test_order import assert_contract_order
 
+try:
+    from unittest.mock import AsyncMock, patch  # Python 3.8+
+except ImportError:
+    from asynctest import CoroutineMock as AsyncMock, patch  # Python 3.7
+
 
 @pytest.mark.asyncio()
 async def test_ws_futures_get_order_book(futuresClientAsync):