refine test (#67959)

RuohengMa · web-flow · commit 05198c6ffb11 · 2024-09-04T14:42:55.000+08:00
diff --git a/test/xpu/test_eye_op_xpu.py b/test/xpu/test_eye_op_xpu.py
@@ -20,6 +20,7 @@
     create_test_class,
     get_xpu_op_support_types,
 )
+from op_test import convert_float_to_uint16
 from op_test_xpu import XPUOpTest
 
 import paddle
@@ -42,8 +43,12 @@ def setUp(self):
                 'num_rows': self.n,
                 'num_columns': self.m,
             }
-            result = np.eye(self.n, self.m, dtype=self.dtype)
-            self.outputs = {'Out': result}
+            if self.dtype == np.uint16:
+                result = np.eye(self.n, self.m, dtype=np.float32)
+                self.outputs = {'Out': convert_float_to_uint16(result)}
+            else:
+                result = np.eye(self.n, self.m, dtype=self.dtype)
+                self.outputs = {'Out': result}
 
         def init_shape(self):
             self.n = 100
@@ -63,6 +68,21 @@ def init_shape(self):
             self.n = 100
             self.m = 1000
 
+    class TestXPUEyeOp3(TestXPUEyeOp):
+        def init_shape(self):
+            self.n = 99
+            self.m = 101
+
+    class TestXPUEyeOp4(TestXPUEyeOp):
+        def init_shape(self):
+            self.n = 2
+            self.m = 2
+
+    class TestXPUEyeOp5(TestXPUEyeOp):
+        def init_shape(self):
+            self.n = 67
+            self.m = 67
+
 
 support_types = get_xpu_op_support_types('eye')
 for stype in support_types:
