small change for IOSuite

yzhliu · Jan 6, 2016 · e0f1252 · e0f1252
1 parent f3b7a6a
commit e0f1252
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/scala-package/core/src/test/scala/ml/dmlc/mxnet/IOSuite.scala b/scala-package/core/src/test/scala/ml/dmlc/mxnet/IOSuite.scala
@@ -35,13 +35,16 @@ class IOSuite extends FunSuite with BeforeAndAfterAll {
     mnistIter.reset()
     mnistIter.iterNext()
     val label0 = mnistIter.getLabel().toArray
+    val data0 = mnistIter.getData().toArray
     mnistIter.iterNext()
     mnistIter.iterNext()
     mnistIter.iterNext()
     mnistIter.reset()
     mnistIter.iterNext()
     val label1 = mnistIter.getLabel().toArray
+    val data1 = mnistIter.getData().toArray
     assert(label0 === label1)
+    assert(data0 === data1)
   }