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)
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)