Could you use `FunSuite` to be more consistent with the other test suites? E.g. https://github.com/apache/spark/blob/master/core/src/test/scala/org/apache/spark/ui/UISuite.scala