* Moved test functions into one location.
* Renamed test functions and module to be more concisely named.
* Moved TF 2.0 setup from setup methods to main function.

PiperOrigin-RevId: 230776467