def setUp(self, max_depth=5): ## or 4, depending from InfiniteGrammar import grammar self.grammar = grammar self.max_depth = max_depth self.trees = [t for t in grammar.enumerate(self.max_depth)]
def setUp(self): from FiniteGrammar import grammar self.grammar = grammar self.trees = [t for t in grammar.enumerate()]