def test_triu_complex128(self):
a = np.asarray(np.random.rand(4, 4), np.complex128)
a_gpu = gpuarray.to_gpu(a)
l_gpu = linalg.triu(a_gpu)
assert np.allclose(np.triu(a), l_gpu.get())
def test_triu_float64(self):
a = np.asarray(np.random.rand(4, 4), np.float64)
a_gpu = gpuarray.to_gpu(a)
l_gpu = linalg.triu(a_gpu)
assert np.allclose(np.triu(a), l_gpu.get())
