I am following other mllib tests here.