diff --git a/homeassistant/components/monoprice/media_player.py b/homeassistant/components/monoprice/media_player.py index 1b1d9d2adf4..20b2ecebcf4 100644 --- a/homeassistant/components/monoprice/media_player.py +++ b/homeassistant/components/monoprice/media_player.py @@ -1,9 +1,11 @@ """Support for interfacing with Monoprice 6 zone home audio controller.""" import logging +from pymonoprice import get_monoprice +from serial import SerialException import voluptuous as vol -from homeassistant.components.media_player import MediaPlayerDevice, PLATFORM_SCHEMA +from homeassistant.components.media_player import PLATFORM_SCHEMA, MediaPlayerDevice from homeassistant.components.media_player.const import ( SUPPORT_SELECT_SOURCE, SUPPORT_TURN_OFF, @@ -20,6 +22,7 @@ from homeassistant.const import ( STATE_ON, ) import homeassistant.helpers.config_validation as cv + from .const import DOMAIN, SERVICE_RESTORE, SERVICE_SNAPSHOT _LOGGER = logging.getLogger(__name__) @@ -68,9 +71,6 @@ def setup_platform(hass, config, add_entities, discovery_info=None): """Set up the Monoprice 6-zone amplifier platform.""" port = config.get(CONF_PORT) - from serial import SerialException - from pymonoprice import get_monoprice - try: monoprice = get_monoprice(port) except SerialException: diff --git a/tests/components/monoprice/test_media_player.py b/tests/components/monoprice/test_media_player.py index a33b8553908..cb064048d7b 100644 --- a/tests/components/monoprice/test_media_player.py +++ b/tests/components/monoprice/test_media_player.py @@ -174,7 +174,10 @@ class TestMonopriceMediaPlayer(unittest.TestCase): self.hass = tests.common.get_test_home_assistant() self.hass.start() # Note, source dictionary is unsorted! - with mock.patch("pymonoprice.get_monoprice", new=lambda *a: self.monoprice): + with mock.patch( + "homeassistant.components.monoprice.media_player.get_monoprice", + new=lambda *a: self.monoprice, + ): setup_platform( self.hass, {