def test_decorated(self):
collator = Collator()
collator.functions = []
# not found
fn = collator._Collator__decorated(1)
self.assertTrue(fn is None)
# found in list
collator.functions = [1]
fn = collator._Collator__decorated(1)
self.assertEqual(fn, (1, None))
# found in dictionary
collator.functions = {1: {"A": 1}}
fn = collator._Collator__decorated(1)
self.assertEqual(fn, (1, {"A": 1}))
def test_decorated(self):
collator = Collator()
collator.functions = []
# not found
fn = collator._Collator__decorated(1)
self.assertTrue(fn is None)
# found in list
collator.functions = [1]
fn = collator._Collator__decorated(1)
self.assertEqual(fn, (1, None))
# found in dictionary
collator.functions = {1: {'A': 1}}
fn = collator._Collator__decorated(1)
self.assertEqual(fn, (1, {'A': 1}))
