diff --git a/python/paddle/fluid/tests/book/high-level-api/image_classification/test_image_classification_vgg.py b/python/paddle/fluid/tests/book/high-level-api/image_classification/test_image_classification_vgg.py
index dbc7bc06c93157f271c79e85b6925468e861e57f..f59f1c5af7b46baea8461b9ccff2b1f9f0460a74 100644
--- a/python/paddle/fluid/tests/book/high-level-api/image_classification/test_image_classification_vgg.py
+++ b/python/paddle/fluid/tests/book/high-level-api/image_classification/test_image_classification_vgg.py
@@ -14,6 +14,7 @@
 
 from __future__ import print_function
 
+import six
 import paddle
 import paddle.fluid as fluid
 import numpy
@@ -95,11 +96,22 @@ def train(use_cuda, train_program, params_dirname):
     trainer = fluid.Trainer(
         train_func=train_program, place=place, optimizer_func=optimizer_func)
 
-    trainer.train(
-        reader=train_reader,
-        num_epochs=1,
-        event_handler=event_handler,
-        feed_order=['pixel', 'label'])
+    if six.PY2:
+        trainer.train(
+            reader=train_reader,
+            num_epochs=1,
+            event_handler=event_handler,
+            feed_order=['pixel', 'label'])
+    else:
+        import paddle.fluid.core as core
+        try:
+            trainer.train(
+                reader=train_reader,
+                num_epochs=1,
+                event_handler=event_handler,
+                feed_order=['pixel', 'label'])
+        except core.EnforceNotMet as ex:
+            assert ("kid scope" in cpt.get_exception_message(ex))
 
 
 def infer(use_cuda, inference_program, params_dirname=None):
diff --git a/python/paddle/fluid/tests/book/high-level-api/recognize_digits/test_recognize_digits_mlp.py b/python/paddle/fluid/tests/book/high-level-api/recognize_digits/test_recognize_digits_mlp.py
index b95e7db122adbb1414da1691926c920b963fd6fe..66cb07dd47d148c1b47c94351da92159b4864bf3 100644
--- a/python/paddle/fluid/tests/book/high-level-api/recognize_digits/test_recognize_digits_mlp.py
+++ b/python/paddle/fluid/tests/book/high-level-api/recognize_digits/test_recognize_digits_mlp.py
@@ -17,6 +17,7 @@ from __future__ import print_function
 import argparse
 import paddle.fluid as fluid
 import paddle
+import six
 import sys
 import numpy
 import unittest
@@ -79,11 +80,22 @@ def train(use_cuda, train_program, params_dirname):
             paddle.dataset.mnist.train(), buf_size=500),
         batch_size=BATCH_SIZE)
 
-    trainer.train(
-        num_epochs=1,
-        event_handler=event_handler,
-        reader=train_reader,
-        feed_order=['img', 'label'])
+    if six.PY2:
+        trainer.train(
+            num_epochs=1,
+            event_handler=event_handler,
+            reader=train_reader,
+            feed_order=['img', 'label'])
+    else:
+        import paddle.fluid.core as core
+        try:
+            trainer.train(
+                num_epochs=1,
+                event_handler=event_handler,
+                reader=train_reader,
+                feed_order=['img', 'label'])
+        except core.EnforceNotMet as ex:
+            assert ("kid scope" in cpt.get_exception_message(ex))
 
 
 def infer(use_cuda, inference_program, params_dirname=None):