Example #1
0
 def test_int_type(self, dtype):
     with self.assertRaises(TypeError):
         generator._check_and_get_dtype(dtype)
Example #2
0
 def test_float16(self):
     with self.assertRaises(TypeError):
         generator._check_and_get_dtype(numpy.float16)
Example #3
0
 def test_float32_64_type(self, dtype):
     self.assertEqual(generator._check_and_get_dtype(dtype),
                      numpy.dtype(dtype))
Example #4
0
 def test_int_type(self, dtype):
     with self.assertRaises(TypeError):
         generator._check_and_get_dtype(dtype)
Example #5
0
 def test_float16(self):
     with self.assertRaises(TypeError):
         generator._check_and_get_dtype(numpy.float16)
Example #6
0
 def test_float32_64_type(self, dtype):
     self.assertEqual(generator._check_and_get_dtype(dtype), numpy.dtype(dtype))