Imp. Add test cases for Spark