def test_inv3_float32(): np.random.seed(42) Ts = np.random.random((1000, 3, 3)).astype(np.float32) Tinv_np = np.array(list(map(np.linalg.inv, Ts.reshape((-1, 3, 3))))).reshape(Ts.shape) Tinv_blitz = inv3(Ts) assert Tinv_blitz.dtype == np.float32 np.set_printoptions(suppress=True) np.testing.assert_allclose(Tinv_np, Tinv_blitz, rtol=1.e-3)
def test_inv3_multiple(): Ts = np.random.random((154, 7, 3, 3)) Tinv_np = np.array(list(map(np.linalg.inv, Ts.reshape((-1, 3, 3))))).reshape(Ts.shape) Tinv_blitz = inv3(Ts) np.set_printoptions(suppress=True) np.testing.assert_allclose(Tinv_np, Tinv_blitz)
def test_inv3(): T = np.random.random((3, 3)) np.testing.assert_allclose(np.linalg.inv(T), inv3(T))