33from MLC .Log .log import set_logger
44from MLC .matlab_engine import MatlabEngine
55from MLC .mlc_parameters .mlc_parameters import Config
6- from MLC .Common .Lisp_Tree_Expr .Lisp_Tree_Expr import Lisp_Tree_Expr , TreeVisitor
6+ from MLC .Common .Lisp_Tree_Expr .Lisp_Tree_Expr import Lisp_Tree_Expr
77from MLC import config as config_path
88
99import os
@@ -73,30 +73,3 @@ def assertNode(self, node, depth, childs, expr_index):
7373
7474 if not node .is_leaf ():
7575 self .assertEquals (len (node ._nodes ), childs )
76-
77- def test_calculate_subtree_depth (self ):
78- expression = '(root (+ (tanh S0) (- (cos S2) (sin S3)))))'
79- expression_tree = Lisp_Tree_Expr (expression )
80- for node in expression_tree .nodes ():
81- print "%s %s %s" % (node .to_string (), node .get_subtreedepth (), node .get_depth ())
82-
83- print "******"
84-
85- class CalculateSubtreeDepth (TreeVisitor ):
86- def __init__ (self ):
87- self ._nodes = {}
88-
89- def visit_internal_node (self , node ):
90- self ._nodes [node ] = max ([self ._nodes [n ] for n in node ._nodes ])+ 1
91-
92- def visit_leaf_node (self , node ):
93- self ._nodes [node ] = 0
94-
95- expression = '(root (+ (tanh S0) (- (cos S2) (sin S3)))))'
96- expression_tree = Lisp_Tree_Expr (expression )
97- visitor = CalculateSubtreeDepth ()
98- expression_tree .get_root_node ().accept (visitor )
99-
100- for node , depth in visitor ._nodes .iteritems ():
101- print "%s %s %s" % (node .to_string (), depth , node .get_depth ())
102- # self.assertTrue(False)
0 commit comments