View source on GitHub
|
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.