def test_invalid_cases(self):
if self.nrhs is None or self.nrhs == 1:
raise unittest.SkipTest()
ng_a = self.a.reshape(1, self.n, self.n)
with pytest.raises(ValueError):
lapack.gesv(ng_a, self.b)
ng_b = self.b.reshape(1, self.n, self.nrhs)
with pytest.raises(ValueError):
lapack.gesv(self.a, ng_b)
ng_a = cupy.ones((self.n, self.n + 1), dtype=self.dtype)
with pytest.raises(ValueError):
lapack.gesv(ng_a, self.b)
ng_a = cupy.ones((self.n + 1, self.n + 1), dtype=self.dtype)
with pytest.raises(ValueError):
lapack.gesv(ng_a, self.b)
ng_a = cupy.ones(self.a.shape, dtype='i')
with pytest.raises(TypeError):
lapack.gesv(ng_a, self.b)
ng_a = cupy.ones((2, self.n, self.n), dtype=self.dtype, order='F')[0]
with pytest.raises(ValueError):
lapack.gesv(ng_a, self.b)
ng_b = self.b.copy(order='C')
with pytest.raises(ValueError):
lapack.gesv(self.a, ng_b)
def test_gesv(self):
lapack.gesv(self.a, self.b)
cupy.testing.assert_allclose(self.b,
self.x_ref,
rtol=self.tol,
atol=self.tol)