diff --git a/python/dllib/src/test/bigdl/transform/test_image.py b/python/dllib/src/test/bigdl/transform/test_image.py
index <HASH>..<HASH> 100644
--- a/python/dllib/src/test/bigdl/transform/test_image.py
+++ b/python/dllib/src/test/bigdl/transform/test_image.py
@@ -18,6 +18,7 @@ import pytest
 import os
 from bigdl.util.common import *
 from bigdl.transform.vision.image import *
+import tempfile
 
 
 class TestLayer():
@@ -198,6 +199,12 @@ class TestLayer():
         image_frame = ImageFrame.read(self.image_path, self.sc)
         image_frame.get_predict()
 
+    def test_read_write_parquet(self):
+        temp = tempfile.mkdtemp() + "testParquet/"
+        resource_path = os.path.join(os.path.split(__file__)[0], "../resources/pascal")
+        ImageFrame.write_parquet(resource_path, temp, self.sc, 1)
+        read_image_frame = ImageFrame.read_parquet(temp, self.sc)
+
 
 if __name__ == "__main__":
     pytest.main([__file__])
[Enhancement] ImageFrame adding more APIs (#<I>)

* add api for evaluation



* rename method



* fix



* wrapper parquet



* fix api



* fix read issue