Пример #1
0
 def testJaxTypeFromVal(self, jaxtype):
     try:
         val = jaxtype(0)
     except TypeError:
         val = jaxtype.type(0)
     self.assertIs(dtypes._jax_type(*dtypes._dtype_and_weaktype(val)),
                   jaxtype)
Пример #2
0
 def testJaxTypeFromType(self, jaxtype):
     self.assertIs(dtypes._jax_type(jaxtype), jaxtype)
Пример #3
0
 def testJaxTypeFromType(self, jaxtype):
     self.assertIs(dtypes._jax_type(*dtypes._dtype_and_weaktype(jaxtype)),
                   jaxtype)