update with tests

apache · Oct 24, 2018 · 5a74776 · 5a74776
1 parent a372583
commit 5a74776
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/scala-package/core/src/test/scala/org/apache/mxnet/NDArraySuite.scala b/scala-package/core/src/test/scala/org/apache/mxnet/NDArraySuite.scala
@@ -21,6 +21,7 @@ import java.io.File
 import java.util.concurrent.atomic.AtomicInteger
 
 import org.apache.mxnet.NDArrayConversions._
+import org.apache.mxnet.util.Visualize
 import org.scalatest.{BeforeAndAfterAll, FunSuite, Matchers}
 
 import scala.collection.mutable.ArrayBuffer
@@ -87,9 +88,9 @@ class NDArraySuite extends FunSuite with BeforeAndAfterAll with Matchers {
 
   test("test Visualize") {
     var nd = NDArray.ones(Shape(1, 2, 100, 1))
-    nd.visualize
+    Visualize.toString(nd)
     nd = NDArray.ones(Shape(1, 4))
-    nd.visualize
+    Visualize.toString(nd)
   }
 
   test("plus") {