Fix import error

2Fake · Oct 7, 2020 · 73f37a6 · 73f37a6
1 parent 8bd04e9
commit 73f37a6
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/tests/test_device.py b/tests/test_device.py
@@ -6,7 +6,7 @@
 import zeroconf
 
 try:
-    from mock import AsyncMock
+    from unittest.mock import AsyncMock
 except ImportError:
     from asynctest import CoroutineMock as AsyncMock
 

diff --git a/tests/test_deviceapi.py b/tests/test_deviceapi.py
@@ -5,7 +5,7 @@
 from httpx import AsyncClient, Client, Response
 
 try:
-    from mock import AsyncMock
+    from unittest.mock import AsyncMock
 except ImportError:
     from asynctest import CoroutineMock as AsyncMock
 

diff --git a/tests/test_plcnetapi.py b/tests/test_plcnetapi.py
@@ -5,7 +5,7 @@
 from httpx import AsyncClient, Client, Response
 
 try:
-    from mock import AsyncMock
+    from unittest.mock import AsyncMock
 except ImportError:
     from asynctest import CoroutineMock as AsyncMock