cleanup: prefer `UnitTests` context name over `BaseTests`

Just cleaning up and moving to our modern convention for naming
this type of context.

/cc @jcredding