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}))