diff --git a/tests/components/onewire/__init__.py b/tests/components/onewire/__init__.py index 7074bc01a7c..561e5a92a23 100644 --- a/tests/components/onewire/__init__.py +++ b/tests/components/onewire/__init__.py @@ -1,5 +1,4 @@ """Tests for 1-Wire integration.""" -from asynctest.mock import patch from homeassistant.components.onewire.const import ( CONF_MOUNT_DIR, @@ -11,6 +10,7 @@ from homeassistant.components.onewire.const import ( from homeassistant.config_entries import CONN_CLASS_LOCAL_POLL from homeassistant.const import CONF_HOST, CONF_PORT, CONF_TYPE +from tests.async_mock import patch from tests.common import MockConfigEntry diff --git a/tests/components/onewire/test_init.py b/tests/components/onewire/test_init.py index 8df9b1ed2ff..8edb5fa0178 100644 --- a/tests/components/onewire/test_init.py +++ b/tests/components/onewire/test_init.py @@ -1,5 +1,4 @@ """Tests for 1-Wire config flow.""" -from asynctest.mock import patch from pyownet.protocol import ConnError, OwnetError from homeassistant.components.onewire.const import CONF_TYPE_OWSERVER, DOMAIN @@ -13,6 +12,7 @@ from homeassistant.const import CONF_HOST, CONF_PORT, CONF_TYPE from . import setup_onewire_owserver_integration, setup_onewire_sysbus_integration +from tests.async_mock import patch from tests.common import MockConfigEntry