Check and only check the output varibles specified by self.outputs.

Author: Xreki

python/paddle/v2/framework/tests/op_test.py

@@ -192,6 +192,9 @@ def check_output_with_place(self, place):
         self.op.run(self.scope, ctx)
 
         for out_name, out_dup in Operator.get_op_outputs(self.op.type()):
+            if out_name not in self.outputs:
+                continue
+
             if out_dup:
                 sub_out = self.outputs[out_name]
                 if not isinstance(sub_out, list):
@@ -206,14 +209,12 @@ def check_output_with_place(self, place):
                             actual, expect, atol=1e-05),
                         "output name: " + out_name + " has diff")
             else:
-                var = self.scope.find_var(out_name)
-                if var is not None:
-                    actual = np.array(var.get_tensor())
-                    expect = self.outputs[out_name]
-                    self.assertTrue(
-                        np.allclose(
-                            actual, expect, atol=1e-05),
-                        "output name: " + out_name + " has diff")
+                actual = np.array(self.scope.find_var(out_name).get_tensor())
+                expect = self.outputs[out_name]
+                self.assertTrue(
+                    np.allclose(
+                        actual, expect, atol=1e-05),
+                    "output name: " + out_name + " has diff")
 
     def check_output(self):
         places = [core.CPUPlace()]