def test_fails_with_mismatched_container_type(self): with golden.check_raises_traceback( 'returns_type_container_mismatch_traceback.expected', TypeError): # This test fails because it `check_returns_type` with a `tuple`, # but returns a `list`. @computation_wrapper_instances.tensorflow_wrapper(tf.int32) @computation_wrapper_instances.check_returns_type((tf.int32, tf.int32)) def _(a): return [a, a]
def test_stackframes_in_errors(self): class DummyError(RuntimeError): pass with golden.check_raises_traceback( 'tensorflow_wrapper_traceback.expected', DummyError): @computation_wrapper_instances.tensorflow_wrapper def _(): raise DummyError()
def test_stackframes_in_errors(self): class DummyError(RuntimeError): pass with golden.check_raises_traceback( 'federated_computation_traceback.expected', DummyError): @federated_computation.federated_computation def _(): raise DummyError()
def test_check_raises_traceback(self): with golden.check_raises_traceback( 'test_check_raises_traceback.expected', RuntimeError): raise RuntimeError()
def test_constructor_argument_normalization_error(self): with golden.check_raises_traceback( 'constructor_argument_normalization_error.expected', TypeError): computation_types.TensorType()
def test_error_on_non_callable_non_type(self): with golden.check_raises_traceback( 'non_callable_non_type_traceback.expected', TypeError): computation_wrapper_instances.tensorflow_wrapper(5)
def test_error_on_non_callable_non_type(self): with golden.check_raises_traceback( 'non_callable_non_type_traceback.expected', TypeError): tensorflow_computation.tf_computation(5)