def testNumValues(self, dtype):
     spec = array_spec.BoundedArraySpec(tuple(),
                                        dtype,
                                        minimum=0,
                                        maximum=9)
     num_values = spec.num_values
     if array_spec.is_discrete(spec):
         self.assertEqual(10, num_values)
     else:
         self.assertEqual(None, num_values)
Exemplo n.º 2
0
def is_discrete(spec):
    if isinstance(spec, TensorSpec):
        return spec.dtype.is_integer
    else:
        return array_spec.is_discrete(spec)