fixes unittests of MPC controller

blobquiet · Mar 28, 2023 · 23992cf · 23992cf
1 parent a7ca773
commit 23992cf
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_mpc_controller.py b/tests/test_mpc_controller.py
@@ -42,7 +42,7 @@ def test_mpc_controller_02_exec(self):
 
         with patch.object(fedbiomed.common.mpc_controller, "subprocess") as mock_process:
             mock_process.Popen.return_value.returncode = 0
-            mock_process.Popen.return_value.communicate.return_value = "Opps", False
+            mock_process.Popen.return_value.communicate.return_value = b"Opps", False
             status = self.mpc_controller._exec(["shamir-server-key"])
             self.assertTrue(status)
 
@@ -64,7 +64,7 @@ def test_mpc_controller_02_exec_shamir(self):
         # Test invalid num parties type
         with patch.object(fedbiomed.common.mpc_controller, "subprocess") as mock_process:
             mock_process.Popen.return_value.returncode = 0
-            mock_process.Popen.return_value.communicate.return_value = "Opps", True
+            mock_process.Popen.return_value.communicate.return_value = b"Opps", True
             result = self.mpc_controller.exec_shamir(party_number=0,
                                                      num_parties=3,
                                                      ip_addresses="dummy/path/to/files")