Merge pull request #147 from natekspencer/dev

Revert "Fix cyclic import linting error"

tests/test_devices.py

@@ -0,0 +1,9 @@
+"""Test the devices."""
+from vivintpy.devices import get_device_class
+from vivintpy.devices.switch import BinarySwitch
+from vivintpy.enums import DeviceType
+
+
+def test_get_device_class() -> None:
+    """Test get device class."""
+    assert get_device_class(DeviceType.BINARY_SWITCH.value) == BinarySwitch

vivintpy/devices/__init__.py

@@ -18,18 +18,20 @@
 
 if TYPE_CHECKING:
     from .alarm_panel import AlarmPanel
+
+DEVICE = "device"
+
+
+def get_device_class(device_type: str) -> Type[VivintDevice]:
+    """Map a device_type string to the class that implements that device."""
+    # pylint: disable=import-outside-toplevel,cyclic-import
     from .camera import Camera
     from .door_lock import DoorLock
     from .garage_door import GarageDoor
     from .switch import BinarySwitch, MultilevelSwitch
     from .thermostat import Thermostat
     from .wireless_sensor import WirelessSensor
 
-DEVICE = "device"
-
-
-def get_device_class(device_type: str) -> Type[VivintDevice]:
-    """Map a device_type string to the class that implements that device."""
     mapping: dict[DeviceType, Type[VivintDevice]] = {
         DeviceType.BINARY_SWITCH: BinarySwitch,
         DeviceType.CAMERA: Camera,