def test_rand_invalid_argument(self):
with self.assertRaises(TypeError):
random.rand(1, 2, 3, unnecessary='unnecessary_argument')
def test_rand(self):
with mock.patch('cupy.random.sample_.random_sample') as m:
random.rand(1, 2, 3, dtype=numpy.float32)
m.assert_called_once_with(size=(1, 2, 3), dtype=numpy.float32)
def test_rand_default_dtype(self):
with mock.patch('cupy.random.sample_.random_sample') as m:
random.rand(1, 2, 3)
m.assert_called_once_with(size=(1, 2, 3), dtype=float)
def test_rand_invalid_argument(self):
with self.assertRaises(TypeError):
random.rand(1, 2, 3, unnecessary='unnecessary_argument')
def test_rand_default_dtype(self):
random.sample_.random_sample = mock.Mock()
random.rand(1, 2, 3)
random.sample_.random_sample.assert_called_once_with(
size=(1, 2, 3), dtype=float)
def test_rand(self):
random.sample_.random_sample = mock.Mock()
random.rand(1, 2, 3, dtype=numpy.float32)
random.sample_.random_sample.assert_called_once_with(
size=(1, 2, 3), dtype=numpy.float32)
def test_rand_default_dtype(self):
random.sample_.random_sample = mock.Mock()
random.rand(1, 2, 3)
random.sample_.random_sample.assert_called_once_with(
size=(1, 2, 3), dtype=float)
def test_rand(self):
random.sample_.random_sample = mock.Mock()
random.rand(1, 2, 3, dtype=numpy.float32)
random.sample_.random_sample.assert_called_once_with(
size=(1, 2, 3), dtype=numpy.float32)