def test_gpu0(self): environment.init_gpu(0) self.assertEqual(environment.array_module(), chainer.cuda.cupy)