Add a test of repeatedly toString calls

ThoughtWorksInc · Feb 11, 2018 · 7314d1a · 7314d1a
1 parent 6baefba
commit 7314d1a
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/Tensors/src/test/scala/com/thoughtworks/compute/TensorsSpec.scala b/Tensors/src/test/scala/com/thoughtworks/compute/TensorsSpec.scala
@@ -26,6 +26,16 @@ class TensorsSpec extends AsyncFreeSpec with Matchers {
         }
       ))
 
+  "repeatedly toString" in {
+    doTensors.map { tensors =>
+      val tensor = tensors.Tensor(42.0f)
+      for (i <- 0 until 1000) {
+        tensor.toString should be("42.0")
+      }
+      succeed
+    }
+  }.run.toScalaFuture
+
   "create a tensor of a constant" in {
     doTensors.flatMap { tensors =>
       val shape = Array(2, 3, 5)