Пример #1
0
 def test_rand_invalid_argument(self):
     with self.assertRaises(TypeError):
         random.rand(1, 2, 3, unnecessary='unnecessary_argument')
Пример #2
0
 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)
Пример #3
0
 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)
Пример #4
0
 def test_rand_invalid_argument(self):
     with self.assertRaises(TypeError):
         random.rand(1, 2, 3, unnecessary='unnecessary_argument')
Пример #5
0
 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)
Пример #6
0
 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)
Пример #7
0
 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)
Пример #8
0
 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)