updated readme

Author: yuanchi2807

codeflare/pipelines/tests/README.md

@@ -1,16 +1,21 @@
-# Architecture decision record
-
-Select a test framework for Codeflare pipeline.
+# Codeflare pipeline tests
 
 Contents:
 
-* [Use pytest as test framework](#use-pytest-as-test-framework)
+* [Unit test execution](#unit-test-execution)
 * [Unit test coverage](#unit-test-coverage)
+* [Use pytest as test framework](#use-pytest-as-test-framework)
 
-## Use pytest as test framework
-PyTest is a testing framework in Python, with simple and easy syntax targeting unit tests and simple functional tests. PyTest can run tests in parallel and automatically detects tests in the test folder. PyTest serves the current goal of testing Codeflare pipelines well.
+## Unit test execution
+Run pytest to execute all test code in the folder
+```
+pytest
+```
 
 ## Unit test coverage
 * and (fan-in) node in a pipeline graph, and variants
 * or (fan-out) node in a pipeline graph, and variants
 * multibranch with mixtures of and/or nodes in a pipeline graph
+
+## Use pytest as test framework
+PyTest is a testing framework in Python, with simple and easy syntax targeting unit tests and simple functional tests. PyTest can run tests in parallel and automatically detects tests in the test folder. PyTest serves the current goal of testing Codeflare pipelines well.

codeflare/pipelines/tests/test_and.py

@@ -20,7 +20,7 @@ def transform(self, xy_list):
         for xy in xy_list:
             X_list.append(xy.get_x())
             y_vec = xy.get_y()
-        X_concat = np.concatenate(X_list, axis=0)
+        X_concat = np.concatenate(X_list, axis=1)
 
         return Xy(X_concat, y_vec)
 
@@ -67,8 +67,8 @@ def test_two_tier_and():
     Xout = ray.get(node_g_output[0].get_Xref())
     yout = ray.get(node_g_output[0].get_yref())
 
-    assert Xout.shape == (40000, 4)
-    assert yout.shape == (10000, 1)
+    assert Xout.shape[0] == 10000
+    assert yout.shape[0] == 10000
 
     ray.shutdown()
 
@@ -111,8 +111,8 @@ def test_four_input_and():
     Xout = ray.get(node_e_output[0].get_Xref())
     yout = ray.get(node_e_output[0].get_yref())
 
-    assert Xout.shape == (40000, 4)
-    assert yout.shape == (10000, 1)
+    assert Xout.shape[0] == 10000
+    assert yout.shape[0] == 10000
 
     ray.shutdown()
 
@@ -148,8 +148,8 @@ def test_two_input_and():
     Xout = ray.get(node_c_output[0].get_Xref())
     yout = ray.get(node_c_output[0].get_yref())
 
-    assert Xout.shape == (20000, 4)
-    assert yout.shape == (10000, 1)
+    assert Xout.shape[0] == 10000
+    assert yout.shape[0] == 10000
 
     ray.shutdown()