def test_clear(self): module = object() imports.post_load_hooks['foo'] = [] imports.run_hooks('foo', module) self.assertFalse('foo' in imports.post_load_hooks)
def test_iterate(self): module = object() self.executed = [] def foo(mod): self.assertTrue(module is mod) self.executed.append('foo') def bar(mod): self.assertTrue(module is mod) self.executed.append('bar') imports.post_load_hooks['foo'] = [foo, bar] imports.run_hooks('foo', module) self.assertEquals(self.executed, ['foo', 'bar'])