[SPARK-3418][MLlib] Matrix unit tests expanded with indexing and updating