kev-inn commented on a change in pull request #893:
URL: https://github.com/apache/systemml/pull/893#discussion_r416600131



##########
File path: 
src/test/java/org/apache/sysds/test/functions/federated/FederatedConstructionTest.java
##########
@@ -62,28 +70,59 @@ public void setUp() {
        }
 
        @Test
-       public void federatedConstructionCP() {
-               federatedConstruction(Types.ExecMode.SINGLE_NODE);
+       public void federatedMatrixConstructionCP() {
+               federatedMatrixConstruction(Types.ExecMode.SINGLE_NODE);
+       }
+
+       @Test
+       public void federatedMatrixConstructionSP() {
+               federatedMatrixConstruction(Types.ExecMode.SPARK);
        }
 
+       public void federatedMatrixConstruction(Types.ExecMode execMode) {
+               getAndLoadTestConfiguration(TEST_NAME);
+               // write input matrix
+               double[][] A = getRandomMatrix(rows, cols, -1, 1, 1, 1234);
+               writeInputMatrixWithMTD("A", A, false, new 
MatrixCharacteristics(rows, cols, blocksize, rows * cols));
+               federatedConstruction(execMode, MATRIX_TEST_FILE_NAME, "A", 
null);
+       }
+       
        @Test
-       public void federatedConstructionSP() {
-               federatedConstruction(Types.ExecMode.SPARK);
+       public void federatedFrameConstructionCP() throws IOException {
+               federatedFrameConstruction(Types.ExecMode.SINGLE_NODE);
+       }
+       
+       /* like other federated functionality, SPARK execution mode is not yet 
working (waiting for better integration

Review comment:
       This is a rather large change and not specific to federated 
construction, it is clear that a better integration is necessary in the future 
so I am not sure if a TODO is helpful or confusing.




----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

For queries about this service, please contact Infrastructure at:
[email protected]


Reply via email to