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"