diff --git a/livy/src/test/java/org/apache/zeppelin/livy/LivyInterpreterIT.java b/livy/src/test/java/org/apache/zeppelin/livy/LivyInterpreterIT.java
index 0c8bea936fc..c7fbc8d4896 100644
--- a/livy/src/test/java/org/apache/zeppelin/livy/LivyInterpreterIT.java
+++ b/livy/src/test/java/org/apache/zeppelin/livy/LivyInterpreterIT.java
@@ -750,6 +750,26 @@ public void testSharedInterpreter() throws InterpreterException {
         assertEquals(1, result.message().size());
         assertTrue(result.message().get(0).getData().contains("col_1 col_2\n1 hello    20"));
       }
+
+      // test plotting of python
+      result = pysparkInterpreter.interpret(
+          "import matplotlib.pyplot as plt\n" +
+          "plt.switch_backend('agg')\n" +
+          "data=[1,2,3,4]\n" +
+          "plt.figure()\n" +
+          "plt.plot(data)\n" +
+          "%matplot plt", context);
+      assertEquals(InterpreterResult.Code.SUCCESS, result.code());
+      assertEquals(1, result.message().size());
+      assertEquals(InterpreterResult.Type.IMG, result.message().get(0).getType());
+
+      // test plotting of R
+      result = sparkRInterpreter.interpret(
+          "hist(mtcars$mpg)", context);
+      assertEquals(InterpreterResult.Code.SUCCESS, result.code());
+      assertEquals(1, result.message().size());
+      assertEquals(InterpreterResult.Type.IMG, result.message().get(0).getType());
+
     } finally {
       sparkInterpreter.close();
       sqlInterpreter.close();