diff --git a/tests/src/test/scala/runtime/actionContainers/Python3AiActionContainerTests.scala b/tests/src/test/scala/runtime/actionContainers/Python3AiActionContainerTests.scala
index 09a1e43..42cdec6 100644
--- a/tests/src/test/scala/runtime/actionContainers/Python3AiActionContainerTests.scala
+++ b/tests/src/test/scala/runtime/actionContainers/Python3AiActionContainerTests.scala
@@ -20,6 +20,7 @@
 import org.junit.runner.RunWith
 import org.scalatest.junit.JUnitRunner
 import common.WskActorSystem
+import spray.json._
 
 //import actionContainers.{ActionContainer}
 //import actionContainers.ActionContainer.withContainer
@@ -29,6 +30,33 @@
 
   override lazy val imageName = "python3aiaction"
 
-  /** indicates if strings in python are unicode by default */
-  override lazy val pythonStringAsUnicode = true
+  it should "run tensorflow" in {
+    val (out, err) = withActionContainer() { c =>
+      val code = """
+                |import tensorflow as tf
+                |def main(args):
+                |    # Initialize two constants
+                |   x1 = tf.constant([1,2,3,4])
+                |   x2 = tf.constant([5,6,7,8])
+                |
+                |   # Multiply
+                |   result = tf.multiply(x1, x2)
+                |
+                |   # Initialize Session and run `result`
+                |   with tf.Session() as sess:
+                |       output = sess.run(result)
+                |       print(output)
+                |       return { "response": output.tolist() }
+            """.stripMargin
+
+      val (initCode, res) = c.init(initPayload(code))
+      initCode should be(200)
+
+      val (runCode, runRes) = c.run(runPayload(JsObject()))
+      runCode should be(200)
+
+      runRes shouldBe defined
+      runRes should be(Some(JsObject("response" -> JsArray(Vector(JsNumber(5),JsNumber(12),JsNumber(21),JsNumber(32))))))
+    }
+  }
 }