def test_execute_tree(): print("Begin test_execute_tree\n") tree_test = Tree() tree_test.add_prep_node([], "denoise", None, None, None) tree_test.add_split_node(["denoise"], "ts2db") tree_test.add_visualize_node(["denoise"], "box_plot") tree_test.add_model_node(["denoise", "ts2db"], "rf") tree_test.add_eval_node(["denoise", "ts2db", "rf"], "MSE") input_file = '../TestData/4_irradiance_train.csv' print(tree_test.execute_tree(input_file))
def test_replicate_subtree(): print("Begin test_replicate_subtree\n") tree_test = Tree() tree_test.add_prep_node([], "denoise", None, None, None) tree_test.add_split_node(["denoise"], "ts2db") tree_test.add_visualize_node(["denoise"], "box_plot") tree_test.add_model_node(["denoise", "ts2db"], "rf") tree_test.add_eval_node(["denoise", "ts2db", "rf"], "MSE") rep = tree_test.replicate_subtree(["denoise", "ts2db", "rf"]) assert rep.root.op == "rf" assert rep.root.children[0].op == "MSE"
def test_all_node_types(): print("Begin test_all_node_types\n") tree_test = Tree() tree_test.add_prep_node([], "denoise", None, None, None) tree_test.add_split_node(["denoise"], "ts2db") tree_test.add_visualize_node(["denoise"], "box_plot") tree_test.add_model_node(["denoise", "ts2db"], "rf") tree_test.add_eval_node(["denoise", "ts2db", "rf"], "MSE") assert tree_test.root.op == "denoise" assert tree_test.root.children[0].op == "ts2db" assert tree_test.root.children[1].op == "box_plot" assert tree_test.root.children[0].children[0].op == "rf" assert tree_test.root.children[0].children[0].children[0].op == "MSE"
def test_load_save(): print("Begin test_load_save\n") tree_test = Tree() tree_test.add_prep_node([], "denoise", None, None, None) tree_test.add_split_node(["denoise"], "ts2db") tree_test.add_visualize_node(["denoise"], "box_plot") tree_test.add_model_node(["denoise", "ts2db"], "rf") tree_test.add_eval_node(["denoise", "ts2db", "rf"], "MSE") SL.save_tree(tree_test, "testingtree") tree_loaded = SL.load_tree("testingtree") assert tree_loaded.root.op == "denoise" assert tree_loaded.root.children[0].op == "ts2db" assert tree_loaded.root.children[1].op == "box_plot" assert tree_loaded.root.children[0].children[0].op == "rf" assert tree_loaded.root.children[0].children[0].children[0].op == "MSE"