[systemds] branch master updated: [MINOR] Fix stability of FederatedRCBindTest

[baunsgaard]

This is an automated email from the ASF dual-hosted git repository.

baunsgaard pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/systemds.git


The following commit(s) were added to refs/heads/master by this push:
     new 3ddf816  [MINOR] Fix stability of FederatedRCBindTest
3ddf816 is described below

commit 3ddf8160fd3dc027825c35b76992afc11f549d96
Author: baunsgaard <baunsga...@tugraz.at>
AuthorDate: Mon Sep 13 21:36:44 2021 +0200

    [MINOR] Fix stability of FederatedRCBindTest
---
 .../functions/federated/primitives/FederatedRCBindTest.java | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git 
a/src/test/java/org/apache/sysds/test/functions/federated/primitives/FederatedRCBindTest.java
 
b/src/test/java/org/apache/sysds/test/functions/federated/primitives/FederatedRCBindTest.java
index cdcb408..1470274 100644
--- 
a/src/test/java/org/apache/sysds/test/functions/federated/primitives/FederatedRCBindTest.java
+++ 
b/src/test/java/org/apache/sysds/test/functions/federated/primitives/FederatedRCBindTest.java
@@ -80,7 +80,6 @@ public class FederatedRCBindTest extends AutomatedTestBase {
        }
 
        @Test
-       @Ignore
        public void federatedRCBindSP() {
                federatedRCBind(Types.ExecMode.SPARK);
        }
@@ -109,8 +108,12 @@ public class FederatedRCBindTest extends AutomatedTestBase 
{
 
                int port1 = getRandomAvailablePort();
                int port2 = getRandomAvailablePort();
+               int port3 = getRandomAvailablePort();
+               int port4 = getRandomAvailablePort();
                Thread t1 = startLocalFedWorkerThread(port1, FED_WORKER_WAIT_S);
                Thread t2 = startLocalFedWorkerThread(port2);
+               Thread t3 = startLocalFedWorkerThread(port3);
+               Thread t4 = startLocalFedWorkerThread(port4);
 
                // we need the reference file to not be written to hdfs, so we 
get the correct format
                rtplatform = Types.ExecMode.SINGLE_NODE;
@@ -134,9 +137,9 @@ public class FederatedRCBindTest extends AutomatedTestBase {
                fullDMLScriptName = HOME + TEST_NAME + ".dml";
                programArgs = new String[] {"-nvargs",
                        "in_A1=" + TestUtils.federatedAddress(port1, 
input("A1")),
-                       "in_A2=" + TestUtils.federatedAddress(port1, 
input("A2")),
-                       "in_B1=" + TestUtils.federatedAddress(port2, 
input("B1")),
-                       "in_B2=" + TestUtils.federatedAddress(port2, 
input("B2")),
+                       "in_A2=" + TestUtils.federatedAddress(port2, 
input("A2")),
+                       "in_B1=" + TestUtils.federatedAddress(port3, 
input("B1")),
+                       "in_B2=" + TestUtils.federatedAddress(port4, 
input("B2")),
                        "in_partitioned=" + 
Boolean.toString(partitioned).toUpperCase(),
                        "in_B1_local=" + input("B1"), "in_B2_local=" + 
input("B2"), "rows=" + rows, "cols=" + cols,
                        "out_R_FF=" + output("R_FF"), "out_R_FL=" + 
output("R_FL"), "out_R_LF=" + output("R_LF"),
@@ -147,7 +150,7 @@ public class FederatedRCBindTest extends AutomatedTestBase {
                // compare all sums via files
                compareResults(1e-11);
 
-               TestUtils.shutdownThreads(t1, t2);
+               TestUtils.shutdownThreads(t1, t2, t3, t4);
                rtplatform = platformOld;
                DMLScript.USE_LOCAL_SPARK_CONFIG = sparkConfigOld;
        }