def test_ijit_returns_same_function_with_args(): def expected_foo(): return True foo = _jit.ijit(1)(expected_foo) assert foo is expected_foo