def test_on_load_decorator_works(self): ctx = Context(env={}) params = [] @ctx.run_on_load def my_on_load(ctx, path): params.append((ctx, path)) self.assertEqual(ctx.on_load_functions, [my_on_load]) ctx.run_on_load_functions() self.assertEqual(params, [(ctx, os.path.abspath(__file__))])