def class_op(x, y):
result = op(_DoubleDouble(x), _DoubleDouble(y))
if isinstance(result, _DoubleDouble):
result = result.to_array()
return result
def testClassInstantiation(self, dtype, val):
dtype = dtypes.canonicalize_dtype(dtype).type
self.assertEqual(dtype(val), _DoubleDouble(val, dtype).to_array())
def testClassUnaryOp(self, dtype, shape, op):
rng = jtu.rand_default(self.rng())
args = (rng(shape, dtype), )
class_op = lambda x: op(_DoubleDouble(x)).to_array()
self.assertAllClose(op(*args), class_op(*args))
def testClassInstantiation(self, dtype, val):
dtype = jnp.dtype(dtype).type
self.assertEqual(dtype(val), _DoubleDouble(val, dtype).to_array())