Make test_ad work with numba.test()