Add count parameter to Scala unit tests.