def test_2d_double(): a = np.finfo(np.float64).tiny * np.abs(np.random.randn(20, 20)) aa = np.copy(a) ftz(aa) bb = np.copy(a) _ftz_numpy(bb) np.testing.assert_array_equal(aa, bb)
def test_2d_single(): a = np.finfo(np.float32).tiny * np.abs(np.random.randn(20, 20)) a = np.asarray(a, dtype=np.float32) aa = np.copy(a) ftz(aa) bb = np.copy(a) _ftz_numpy(bb) np.testing.assert_array_equal(aa, bb)