ugh

Author: rickwierenga

pylabrobot/centrifuge/centrifuge.py

@@ -145,12 +145,7 @@ async def unload(self) -> None:  # DOOR arg?
 
     await self.backend.unload()
 
-    print("assigning to self:", self.centrifuge.at_bucket.resource)
     self.assign_child_resource(self.centrifuge.at_bucket.resource)
-    print("assigned to self:", self.resource)
-    print("paren :", self.resource.parent)
-    print(" .           :", self.centrifuge.at_bucket)
-    print(" .           :", self.centrifuge.at_bucket.resource)
 
   def serialize(self) -> dict:
     return {
@@ -164,7 +159,7 @@ def serialize(self) -> dict:
   def deserialize(cls, data: dict, allow_marshall: bool = False):
     data_copy = data.copy()  # copy data because we will be modifying it
     centrifuge_data = data_copy.pop("centrifuge")
-    centrifuge = Centrifuge.deserialize(centrifuge_data, allow_marshall=allow_marshall)
+    centrifuge = Centrifuge.deserialize(centrifuge_data)
     return cls(
       centrifuge=centrifuge,
       **data_copy,

pylabrobot/centrifuge/centrifuge_tests.py

@@ -45,6 +45,7 @@ async def test_load(self):
     self.loader.assign_child_resource(self.plate)
     await self.loader.load()
     self.mock_loader_backend.load.assert_awaited_once()
+    assert self.centrifuge.at_bucket is not None
     self.assertEqual(self.centrifuge.at_bucket.children[0], self.plate)
     self.assertEqual(self.loader.children, [])
 
@@ -64,6 +65,7 @@ async def test_load_no_plate(self):
   async def test_load_bucket_has_plate(self):
     await self.centrifuge.go_to_bucket1()
     await self.centrifuge.open_door()
+    assert self.centrifuge.at_bucket is not None
     self.centrifuge.at_bucket.assign_child_resource(self.plate)
     another_plate = Cor_96_wellplate_360ul_Fb(name="another_plate")
     self.loader.assign_child_resource(another_plate)
@@ -81,6 +83,7 @@ async def test_load_not_at_bucket(self):
   async def test_unload(self):
     await self.centrifuge.go_to_bucket1()
     await self.centrifuge.open_door()
+    assert self.centrifuge.at_bucket is not None
     self.centrifuge.at_bucket.assign_child_resource(self.plate)
     await self.loader.unload()
     self.mock_loader_backend.unload.assert_awaited_once()