asserting for number of terminal nodes

Author: msrivats@us.ibm.com

codeflare/pipelines/test_Datamodel.py

@@ -77,9 +77,7 @@ def test_multibranch(self):
         pipeline_input.add_xy_arg(node_a, dm.Xy(X_train, y_train))
 
         terminal_nodes = pipeline.get_terminal_nodes()
-        print('Num terminal nodes: ' + str(len(terminal_nodes)))
-        for terminal_node in terminal_nodes:
-            print(str(terminal_node))
+        assert len(terminal_nodes) == 4
 
         ## execute the codeflare pipeline
         pipeline_output = rt.execute_pipeline(pipeline, ExecutionType.FIT, pipeline_input)