added asserts to check codeflare pipeline execution

Author: yuanchi2807

codeflare/tests/test_and.py

@@ -54,9 +54,12 @@ def test_and():
 
     ## retrieve node c
     out_Xyrefs = ray.get(out_args[node_c])
+    assert out_Xyrefs
+
     for out_xyref in out_Xyrefs:
         x = ray.get(out_xyref.get_Xref())
         and_func = ray.get(out_xyref.get_currnoderef()).get_and_func()
+        assert x.any()
         print(x)
 
     ray.shutdown()

codeflare/tests/test_or.py

@@ -58,13 +58,15 @@ def test_or():
     in_args={node_a: Xy_ref_ptrs}
     ## execute the codeflare pipeline
     out_args = rt.execute_pipeline(pipeline, ExecutionType.FIT, in_args)
+    assert out_args
 
     ## retrieve node b
     node_b_out_args = ray.get(out_args[node_b])
     b_out_xyref = node_b_out_args[0]
     ray.get(b_out_xyref.get_Xref())
     b_out_node = ray.get(b_out_xyref.get_currnoderef())
     sct_b = b_out_node.get_estimator()
+    assert sct_b
     print(sct_b.feature_importances_)
 
     ray.shutdown()