def test_register_check(self): class TestCheck(policy.Check): pass policy.register('spam', TestCheck) self.assertEqual(policy._checks, dict(spam=TestCheck))
def test_register_func(self): policy.register('spam', 'func') self.assertEqual(policy.Brain._checks, dict(spam='func'))
def test_register_func(self): policy.register("spam", "func") self.assertEqual(policy.Brain._checks, dict(spam="func"))