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))