fix dist unittest

python/paddle/fluid/tests/unittests/test_desc_clone_dist.py

@@ -19,7 +19,7 @@ import unittest
 import paddle
 import paddle.fluid as fluid
 
-from test_desc_clone import get_model
+from test_desc_clone import get_model, program_equal
 
 
 def get_transpiler(trainer_id, main_program, pserver_endpoints, trainers):

python/paddle/fluid/tests/unittests/test_program_code_dist.py

@@ -17,6 +17,7 @@ from __future__ import print_function
 import unittest
 import sys
 
+import paddle
 import paddle.fluid as fluid
 import paddle.fluid.layers as layers
 from paddle.fluid.layers.io import ListenAndServ