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)
