We encourage using scalatest assert statements instead of junit asserts. For example this would be `assert(trainSize === 0.025)`