Module: tff.test

Libraries for testing TensorFlow Federated.

Classes

class TestCase: Base class for TensroFlow Federated tests.

Functions

assert_contains_secure_aggregation(...): Asserts that comp contains at least one secure aggregation call.

assert_contains_unsecure_aggregation(...): Asserts that comp contains at least one unsecure aggregation call.

assert_not_contains_secure_aggregation(...): Asserts that comp contains no secure aggregation calls.

assert_not_contains_unsecure_aggregation(...): Asserts that comp contains no unsecure aggregation calls.

main(...): Runs all unit tests with TF 2.0 features enabled.

set_no_default_context(...): Places a RuntimeErrorContext at the bottom of the stack.