示例#1
0
 def test_int_type(self, dtype):
     with self.assertRaises(TypeError):
         _generator._check_and_get_dtype(dtype)
示例#2
0
 def test_float32_64_type(self, dtype):
     assert _generator._check_and_get_dtype(dtype) == numpy.dtype(dtype)
示例#3
0
 def test_float16(self):
     with self.assertRaises(TypeError):
         _generator._check_and_get_dtype(numpy.float16)
示例#4
0
 def test_float32_64_type(self, dtype):
     self.assertEqual(_generator._check_and_get_dtype(dtype),
                      numpy.dtype(dtype))