def testMultipleExtensions():
"""Test behavior of multiple extensions."""
class Test1ExtensionNode(mdp.ExtensionNode, mdp.Node):
extension_name = "__test1"
def _testtest1(self):
pass
class Test2ExtensionNode(mdp.ExtensionNode, mdp.Node):
extension_name = "__test2"
def _testtest2(self):
pass
mdp.activate_extension("__test1")
node = mdp.Node()
node._testtest1()
mdp.activate_extension("__test2")
node._testtest2()
mdp.deactivate_extension("__test1")
assert not hasattr(mdp.nodes.SFANode, "_testtest1")
mdp.activate_extension("__test1")
node._testtest1()
mdp.deactivate_extensions(["__test1", "__test2"])
assert not hasattr(mdp.nodes.SFANode, "_testtest1")
assert not hasattr(mdp.nodes.SFANode, "_testtest2")
def testExtensionInheritanceTwoExtensions():
"""Test non_extension injection for multiple extensions."""
class Test1ExtensionNode(mdp.ExtensionNode):
extension_name = "__test1"
def _execute(self):
return 1
class Test2ExtensionNode(mdp.ExtensionNode):
extension_name = "__test2"
class Test3ExtensionNode(mdp.ExtensionNode):
extension_name = "__test3"
def _execute(self):
return "3a"
class TestNode1(mdp.Node):
pass
class TestNode2(TestNode1):
pass
class ExtendedTest1Node2(Test1ExtensionNode, TestNode2):
pass
class ExtendedTest2Node1(Test2ExtensionNode, TestNode1):
def _execute(self):
return 2
class ExtendedTest3Node1(Test3ExtensionNode, TestNode1):
def _execute(self):
return "3b"
test_node = TestNode2()
mdp.activate_extension('__test2')
assert test_node._execute() == 2
mdp.deactivate_extension('__test2')
# in this order TestNode2 should get execute from __test1,
# the later addition by __test1 to TestNode1 doesn't matter
mdp.activate_extensions(['__test1', '__test2'])
assert test_node._execute() == 1
mdp.deactivate_extensions(['__test2', '__test1'])
# now activate in inverse order
# TestNode2 already gets _execute from __test2, but that is still
# overriden by __test1, thats how its registered in _extensions
mdp.activate_extensions(['__test2', '__test1'])
assert test_node._execute() == 1
mdp.deactivate_extensions(['__test2', '__test1'])
## now the same with extension 3
mdp.activate_extension('__test3')
assert test_node._execute() == "3b"
mdp.deactivate_extension('__test3')
# __test3 does not override, since the _execute slot for Node2
# was first filled by __test1
mdp.activate_extensions(['__test3', '__test1'])
assert test_node._execute() == 1
mdp.deactivate_extensions(['__test3', '__test1'])
# inverse order
mdp.activate_extensions(['__test1', '__test3'])
assert test_node._execute() == 1
mdp.deactivate_extensions(['__test2', '__test1'])
def testMultipleExtensions():
"""Test behavior of multiple extensions."""
class Test1ExtensionNode(mdp.ExtensionNode, mdp.Node):
extension_name = "__test1"
def _testtest1(self):
pass
class Test2ExtensionNode(mdp.ExtensionNode, mdp.Node):
extension_name = "__test2"
def _testtest2(self):
pass
mdp.activate_extension("__test1")
node = mdp.Node()
node._testtest1()
mdp.activate_extension("__test2")
node._testtest2()
mdp.deactivate_extension("__test1")
assert not hasattr(mdp.nodes.SFANode, "_testtest1")
mdp.activate_extension("__test1")
node._testtest1()
mdp.deactivate_extensions(["__test1", "__test2"])
assert not hasattr(mdp.nodes.SFANode, "_testtest1")
assert not hasattr(mdp.nodes.SFANode, "_testtest2")
def teardown_function(function):
"""Deactivate all extensions and remove testing extensions."""
mdp.deactivate_extensions(mdp.get_active_extensions())
for key in mdp.get_extensions().copy():
if key.startswith("__test"):
del mdp.get_extensions()[key]
def setup_environment(self):
"""Activate the used extensions."""
# deactivate all active extensions for safety
mdp.deactivate_extensions(mdp.get_active_extensions())
mdp.activate_extensions(self._used_extensions)
def setup_environment(self):
"""Activate the used extensions."""
# deactivate all active extensions for safety
mdp.deactivate_extensions(mdp.get_active_extensions())
mdp.activate_extensions(self._used_extensions)
def testExtensionInheritanceTwoExtensions():
"""Test non_extension injection for multiple extensions."""
class Test1ExtensionNode(mdp.ExtensionNode):
extension_name = "__test1"
def _execute(self):
return 1
class Test2ExtensionNode(mdp.ExtensionNode):
extension_name = "__test2"
class Test3ExtensionNode(mdp.ExtensionNode):
extension_name = "__test3"
def _execute(self):
return "3a"
class TestNode1(mdp.Node):
pass
class TestNode2(TestNode1):
pass
class ExtendedTest1Node2(Test1ExtensionNode, TestNode2):
pass
class ExtendedTest2Node1(Test2ExtensionNode, TestNode1):
def _execute(self):
return 2
class ExtendedTest3Node1(Test3ExtensionNode, TestNode1):
def _execute(self):
return "3b"
test_node = TestNode2()
mdp.activate_extension('__test2')
assert test_node._execute() == 2
mdp.deactivate_extension('__test2')
# in this order TestNode2 should get execute from __test1,
# the later addition by __test1 to TestNode1 doesn't matter
mdp.activate_extensions(['__test1', '__test2'])
assert test_node._execute() == 1
mdp.deactivate_extensions(['__test2', '__test1'])
# now activate in inverse order
# TestNode2 already gets _execute from __test2, but that is still
# overriden by __test1, thats how its registered in _extensions
mdp.activate_extensions(['__test2', '__test1'])
assert test_node._execute() == 1
mdp.deactivate_extensions(['__test2', '__test1'])
## now the same with extension 3
mdp.activate_extension('__test3')
assert test_node._execute() == "3b"
mdp.deactivate_extension('__test3')
# __test3 does not override, since the _execute slot for Node2
# was first filled by __test1
mdp.activate_extensions(['__test3', '__test1'])
assert test_node._execute() == 1
mdp.deactivate_extensions(['__test3', '__test1'])
# inverse order
mdp.activate_extensions(['__test1', '__test3'])
assert test_node._execute() == 1
mdp.deactivate_extensions(['__test2', '__test1'])
