def test_map(self, _decorated, _function, _methods, _classes):
func_globals = {"A": 1, "B": 2}
methods = ["m1", "m2", "m3", "m4", "m5", "m6", "m7", "m8", "m9"]
functions = ["f1", "f2", "f3", "f4", "f5", "f6", "f7", "f8", "f9"]
_classes.return_value = ["c1", "c2", "c3"]
_methods.side_effect = [methods[0:3], methods[3:6], methods[6:9]]
_function.side_effect = functions
_decorated.side_effect = [(fn, None) for fn in functions]
collator = Collator()
collator._Collator__map(func_globals)
_classes.assert_called_with(func_globals)
_methods.assert_has_calls([call(c) for c in _classes.return_value])
_function.assert_has_calls([call(m) for m in methods])
self.assertEqual(collator.bound, set(functions))
def test_map(self, _decorated, _function, _methods, _classes):
func_globals = {'A': 1, 'B': 2}
methods = [
'm1',
'm2',
'm3',
'm4',
'm5',
'm6',
'm7',
'm8',
'm9'
]
functions = [
'f1',
'f2',
'f3',
'f4',
'f5',
'f6',
'f7',
'f8',
'f9'
]
_classes.return_value = [
'c1',
'c2',
'c3'
]
_methods.side_effect = [
methods[0:3],
methods[3:6],
methods[6:9]
]
_function.side_effect = functions
_decorated.side_effect = [(fn, None) for fn in functions]
collator = Collator()
collator._Collator__map(func_globals)
_classes.assert_called_with(func_globals)
_methods.assert_has_calls([call(c) for c in _classes.return_value])
_function.assert_has_calls([call(m) for m in methods])
self.assertEqual(collator.bound, set(functions))
