Switching to JUnit/ScalaTest to make it easier to run from Gradle