def test_get_transformer(self):
ntm = AstTransformerManager()
ntm.reload()
ntsc = ntm.get_instance_by_name('IdentityTransform')
assert_is_instance(ntsc, IdentityTransform)
class TreeTransformController(object):
def __init__(self):
self.ast_tree_manager = AstTreeManager()
self.ast_transformer_manager = AstTransformerManager()
def clear(self):
self.ast_tree_manager.clear()
self.ast_transformer_manager.clear()
def apply_transform(self, tree=None, transform=None, name=None):
"""
creates a new ast_tree item by applying
transform to tree, updates the controllers
internals to track that
"""
if isinstance(tree, int):
tree = self.ast_tree_manager[tree]
if isinstance(transform, int):
transform = self.ast_transformer_manager[transform]
new_ast_tree = self.ast_tree_manager.create_transformed_child(
tree, transform, name=name
)
return new_ast_tree
def load_transforms(self, key):
self.ast_transformer_manager.load_transforms(key)
def test_get_transformer(self):
ntm = AstTransformerManager()
ntm.reload()
ntsc = ntm.get_instance_by_name('IdentityTransform')
assert_is_instance(ntsc, IdentityTransform)
class TreeTransformController(object):
def __init__(self):
self.ast_tree_manager = AstTreeManager()
self.ast_transformer_manager = AstTransformerManager()
def clear(self):
self.ast_tree_manager.clear()
self.ast_transformer_manager.clear()
def apply_transform(self, tree=None, transform=None, name=None):
"""
creates a new ast_tree item by applying
transform to tree, updates the controllers
internals to track that
"""
if isinstance(tree, int):
tree = self.ast_tree_manager[tree]
if isinstance(transform, int):
transform = self.ast_transformer_manager[transform]
new_ast_tree = self.ast_tree_manager.create_transformed_child(
tree, transform, name=name)
return new_ast_tree
def load_transforms(self, key):
self.ast_transformer_manager.load_transforms(key)
def test_basic_load(self):
ntm = AstTransformerManager()
ntm.get_ast_transformers(
"ast_tool_box.transformers.identity_transform")
ntm.reload()
assert_greater(ntm.count(), 0)
def __init__(self):
self.ast_tree_manager = AstTreeManager()
self.ast_transformer_manager = AstTransformerManager()
def test_basic_load(self):
ntm = AstTransformerManager()
ntm.get_ast_transformers("ast_tool_box.transformers.identity_transform")
ntm.reload()
assert_greater(ntm.count(), 0)
def __init__(self):
self.ast_tree_manager = AstTreeManager()
self.ast_transformer_manager = AstTransformerManager()
