def test_srcedit():
""" Test src edition """
# Change src
pm = PackageManager()
pm.wraleapath = "."
pm.init(".")
factory = pm["TestPackage"]["test"]
node1 = factory.instantiate()
assert node1.func((1, 2, 3)) == (1, 2, 3)
src = factory.get_node_src()
assert src
newsrc = src.replace("return inputs", "return sum(inputs)")
assert newsrc
factory.apply_new_src(newsrc)
node2 = factory.instantiate()
assert node2.func((1, 2, 3)) == 6
factory.save_new_src(newsrc)
src = factory.get_node_src()
print src
return
# Reinit src
pm = PackageManager()
pm.wraleapath = "."
pm.init()
factory = pm["TestPackage"]["test"]
node = factory.instantiate()
assert node(((1, 2, 3),)) == 6
def test_srcedit():
""" Test src edition """
# Change src
pm = PackageManager()
pm.wraleapath = '.'
pm.init('.')
factory = pm['TestPackage']['test']
node1 = factory.instantiate()
assert node1.func((1, 2, 3)) == (1, 2, 3)
src = factory.get_node_src()
assert src
newsrc = src.replace("return inputs", "return sum(inputs)")
assert newsrc
factory.apply_new_src(newsrc)
node2 = factory.instantiate()
assert node2.func((1, 2, 3)) == 6
factory.save_new_src(newsrc)
src = factory.get_node_src()
print src
return
# Reinit src
pm = PackageManager()
pm.wraleapath = '.'
pm.init()
factory = pm['TestPackage']['test']
node = factory.instantiate()
assert node(((1, 2, 3), )) == 6
