You can also look at spark-testing-base which works in both Scalatest and Junit and see if that works for your use case.

On Friday, April 1, 2016, Ted Yu <yuzhihong@gmail.com> wrote:
Assuming your code is written in Scala, I would suggest using ScalaTest.

Please take a look at the XXSuite.scala files under mllib/

On Fri, Apr 1, 2016 at 1:31 PM, Shishir Anshuman <shishiranshuman@gmail.com> wrote:
Hello,

I have a code written in scala using Mllib. I want to perform unit testing it. I cant decide between Junit 4 and ScalaTest. 
I am new to Spark. Please guide me how to proceed with the testing.

Thank you.



--
Cell : 425-233-8271
Twitter: https://twitter.com/holdenkarau