def test_functions(self, _decorated, _getmodule):
m1 = Module("m1")
m2 = Module("m2")
modules = [m1, m2, m2]
_decorated.side_effect = lambda x: (x, None)
_getmodule.side_effect = modules
collator = Collator()
for n in range(4):
name = "f%d" % n
fn = Mock(name=name)
fn.__name__ = name
collator.functions.append(fn)
collator.bound.add(collator.functions[1])
functions = collator._Collator__functions()
for module, fns in functions.items():
if module == m1:
self.assertTrue(hasattr(module, "f0"))
self.assertEqual(fns, [(f, None) for f in collator.functions[0:1]])
continue
if module == m2:
self.assertTrue(hasattr(module, "f2"))
self.assertTrue(hasattr(module, "f3"))
self.assertEqual(fns, [(f, None) for f in collator.functions[2:4]])
continue
def test_functions(self, _decorated, _getmodule):
m1 = Module('m1')
m2 = Module('m2')
modules = [m1, m2, m2]
_decorated.side_effect = lambda x: (x, None)
_getmodule.side_effect = modules
collator = Collator()
for n in range(4):
name = 'f%d' % n
fn = Mock(name=name)
fn.__name__ = name
collator.functions.append(fn)
collator.bound.add(collator.functions[1])
functions = collator._Collator__functions()
for module, fns in functions.items():
if module == m1:
self.assertTrue(hasattr(module, 'f0'))
self.assertEqual(fns, [(f, None) for f in collator.functions[0:1]])
continue
if module == m2:
self.assertTrue(hasattr(module, 'f2'))
self.assertTrue(hasattr(module, 'f3'))
self.assertEqual(fns, [(f, None) for f in collator.functions[2:4]])
continue
