Project import generated by Copybara.

Migrate basic coarse level e2e Blueberry tests.

Test: Manual test
PiperOrigin-RevId: 416413362
Change-Id: I97989892034864836a39c758936fb31d6caea763

1 files changed, 4 insertions, 1 deletions

diff --git a/system/blueberry/utils/arduino_base.py b/system/blueberry/utils/arduino_base.py
index 37c9146033..e84f2e6892 100644
--- a/system/blueberry/utils/arduino_base.py
+++ b/system/blueberry/utils/arduino_base.py
@@ -7,6 +7,7 @@ Internal link
 """
 
 import time
+from typing import Dict
 from mobly.signals import ControllerError
 import serial
 
@@ -15,13 +16,15 @@ class ArduinoBase(object):
   """Implements an Arduino base class.
 
   Attributes:
+    config: A device configuration.
     serial: serial object, a serial object which is used to communicate with
       Arduino board.
   """
 
-  def __init__(self, config):
+  def __init__(self, config: Dict[str, str]):
     """Initializes an Arduino base class."""
     self._verify_config(config)
+    self.config = config
     self.serial = serial.Serial(config['arduino_port'], 9600)
     self.serial.timeout = 30
     # Buffer between calling serial.Serial() and serial.Serial.write().