def test_speeds(): # This very superficial test checks that angular_integration is able to # execute (no syntax errors) n = 101 IM = np.random.randn(n, n) angular_integration(IM)
def check(name, ref, rtol, IM, **kwargs): with catch_warnings(): simplefilter('ignore', category=DeprecationWarning) r, speeds = vmi.angular_integration(IM, **kwargs) # (ignoring pixels at r = 0 and 1, which can be poor) assert_allclose(speeds[2:R], ref(r[2:R]), rtol=rtol, err_msg='-> {}, {}'.format(name, kwargs))
def test_speeds_non_integer_center(): # ensures that the rest speeds function can work with a non-integer center n = 101 IM = np.random.randn(n, n) angular_integration(IM, origin=(50.5, 50.5))