Factor existing test decorator into smaller reusable components.
Factor existing test decorator into smaller reusable components.

* Enabled `NonDeterministicTest` for all contexts.

PiperOrigin-RevId: 328846758

tensorflow/federated,tensorflow/federated,tensorflow/federated