fix(arduino): stop require optional dependency 'esp'

espressif · Apr 30, 2024 · e7db668 · e7db668
1 parent 1387af5
commit e7db668
Showing 1 changed file with 16 additions and 5 deletions.
diff --git a/pytest-embedded-arduino/pytest_embedded_arduino/__init__.py b/pytest-embedded-arduino/pytest_embedded_arduino/__init__.py
@@ -1,11 +1,22 @@
 """Make pytest-embedded plugin work with Arduino."""
 
+import importlib
+
+from pytest_embedded.utils import lazy_load
+
 from .app import ArduinoApp
-from .serial import ArduinoSerial
 
-__all__ = [
-    'ArduinoApp',
-    'ArduinoSerial',
-]
+__getattr__ = lazy_load(
+    importlib.import_module(__name__),
+    {
+        'ArduinoApp': ArduinoApp,
+    },
+    {
+        'ArduinoSerial': '.serial',  # requires esp
+    },
+)
+
+__all__ = ['ArduinoApp', 'ArduinoSerial']
+
 
 __version__ = '1.10.0'