protoplaster: tests: Add force option to SMBus tests

Signed-off-by: Krzysztof Sychla <ksychla@antmicro.com>

Author: ksychla

protoplaster/tests/i2c_mux/TCA9548A/TCA9548A.py

@@ -4,8 +4,11 @@
 
 class TCA9548A(I2CMux):
 
-    def __init__(self, i2c_bus: int, i2c_address: int):
-        self.bus = SMBus(i2c_bus)
+    def __init__(self,
+                 i2c_bus: int,
+                 i2c_address: int,
+                 smbus_force: bool = False):
+        self.bus = SMBus(i2c_bus, force=smbus_force)
         self.address = i2c_address
 
     def is_alive(self):

protoplaster/tests/i2c_mux/TCA9548A/test.py

@@ -18,7 +18,8 @@ def test_is_alive(self):
         check if TCA9548A responds correctly to simple requests
         {%- endmacro %}
         """
-        mux = TCA9548A(self.i2c_bus, self.i2c_address)
+        self.smbus_force = getattr(self, "smbus_force", False)
+        mux = TCA9548A(self.i2c_bus, self.i2c_address, self.smbus_force)
         assert mux.is_alive(), "TCA9548A does not respond correctly"
 
     def name(self):

protoplaster/tests/pmic/DA9062/DA9062.py

@@ -80,8 +80,8 @@ class LDOSelection(IntEnum):
 class DA9062:
     DEVICE_ID = 0x62
 
-    def __init__(self, i2c_bus, i2c_address):
-        self.bus = SMBus(i2c_bus)
+    def __init__(self, i2c_bus, i2c_address, smbus_force):
+        self.bus = SMBus(i2c_bus, force=smbus_force)
         self.i2c_address = i2c_address
 
     def is_alive(self):

protoplaster/tests/pmic/DA9062/test.py

@@ -13,11 +13,10 @@ class TestDA9062:
     """
 
     def setup_class(self):
-        if not hasattr(self, 'current_selections'):
-            self.current_selections = []
-        if not hasattr(self, 'current_voltages'):
-            self.current_voltages = []
-        self.pmic = DA9062(self.bus, self.address)
+        self.current_selections = getattr(self, "current_selections", [])
+        self.current_voltages = getattr(self, "current_voltages", [])
+        self.smbus_force = getattr(self, "smbus_force", False)
+        self.pmic = DA9062(self.bus, self.address, self.smbus_force)
 
     def test_is_alive(self):
         """

protoplaster/tests/pmic/UCD90320U/UCD90320U.py

@@ -11,8 +11,11 @@ class UCD90320URegs(IntEnum):
 
 class UCD90320U:
 
-    def __init__(self, i2c_bus, i2c_address):
-        self.bus = SMBus(i2c_bus)
+    def __init__(self,
+                 i2c_bus: int,
+                 i2c_address: int,
+                 smbus_force: bool = False):
+        self.bus = SMBus(i2c_bus, smbus_force)
         self.address = i2c_address
 
     def is_alive(self):

protoplaster/tests/pmic/UCD90320U/test.py

@@ -19,7 +19,8 @@ def test_is_alive(self):
           check if UCD90320U responds correctly to simple requests
         {%- endmacro %}
         """
-        pmic = UCD90320U(self.bus, self.address)
+        self.smbus_force = getattr(self, "smbus_force", False)
+        pmic = UCD90320U(self.bus, self.address, self.smbus_force)
         assert pmic.is_alive(), "UCD90320U does not respond correctly"
 
     def name(self):

protoplaster/tests/thermometer/TMP431/TMP431.py

@@ -14,8 +14,11 @@ class TMP431:
     DEVICE_ID = 0x31
     MANUFACTURER_ID = 0x55
 
-    def __init__(self, i2c_bus, i2c_address):
-        self.bus = SMBus(i2c_bus)
+    def __init__(self,
+                 i2c_bus: int,
+                 i2c_address: int,
+                 smbus_force: bool = False):
+        self.bus = SMBus(i2c_bus, force=smbus_force)
         self.address = i2c_address
 
     def is_alive(self):

protoplaster/tests/thermometer/TMP431/test.py

@@ -18,7 +18,8 @@ def test_is_alive(self):
           check if TMP431 responds correctly to simple requests
         {%- endmacro %}
         """
-        thermometer = TMP431(self.bus, self.address)
+        self.smbus_force = getattr(self, "smbus_force", False)
+        thermometer = TMP431(self.bus, self.address, self.smbus_force)
         assert thermometer.is_alive(), f"TMP431 does not respond correctly"
 
     def name(self):