fix atc num of temp zones (#766)

Author: xbtu2

pylabrobot/thermocycling/thermo_fisher/proflex_tests.py

@@ -10,6 +10,7 @@ class TestProflexBackend(unittest.IsolatedAsyncioTestCase):
   async def asyncSetUp(self):
     await super().asyncSetUp()
     self.proflex = ProflexBackend(ip="1.2.3.4")
+    self.proflex.num_temp_zones = 2
     self.proflex.io.write = unittest.mock.AsyncMock()  # type: ignore
     self.proflex.io.read = unittest.mock.AsyncMock()  # type: ignore
     self.proflex.io.read_until_eof = unittest.mock.AsyncMock()  # type: ignore

pylabrobot/thermocycling/thermo_fisher/thermo_fisher_thermocycler.py

@@ -442,7 +442,7 @@ async def _load_num_blocks_and_type(self):
       self.num_temp_zones = 2
     elif self.bid == "31":
       self._num_blocks = 1
-      self.num_temp_zones = 1
+      self.num_temp_zones = 3
     else:
       raise NotImplementedError("Only BID 31, 12 and 13 are supported")
 
@@ -936,6 +936,15 @@ async def run_protocol(
       if isinstance(stage, Step):
         protocol.stages[i] = Stage(steps=[stage], repeats=1)
 
+    for stage in protocol.stages:
+      for step in stage.steps:
+        if len(step.temperature) != self.num_temp_zones:
+          raise ValueError(
+            f"Each step in the protocol must have a list of temperatures "
+            f"of length {self.num_temp_zones}. "
+            f"Step temperatures: {step.temperature} (length {len(step.temperature)})"
+          )
+
     stage_name_prefixes = stage_name_prefixes or ["Stage_" for i in range(len(protocol.stages))]
 
     await self._scpi_write_run_info(