def test_int_type(self, dtype): with self.assertRaises(TypeError): generator._check_and_get_dtype(dtype)
def test_float16(self): with self.assertRaises(TypeError): generator._check_and_get_dtype(numpy.float16)
def test_float32_64_type(self, dtype): self.assertEqual(generator._check_and_get_dtype(dtype), numpy.dtype(dtype))