add unittest

PaddlePaddle · Aug 28, 2023 · d8f8bb5 · d8f8bb5
1 parent 81770a5
commit d8f8bb5
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/test/legacy_test/test_linear_interp_v2_op.py b/test/legacy_test/test_linear_interp_v2_op.py
@@ -337,6 +337,24 @@ def test_case(self):
             np.testing.assert_allclose(interp.numpy(), expect, rtol=1e-05)
 
 
+class TestLinearInterpOpAPI2_0_case2(unittest.TestCase):
+    def test_case(self):
+        # dygraph
+        x_data = np.random.random((1, 3, 128)).astype("float32")
+        us_1 = paddle.nn.Upsample(
+            size=[64], mode='linear', align_mode=1, align_corners=False
+        )
+        with fluid.dygraph.guard():
+            x = fluid.dygraph.to_variable(x_data)
+            interp = us_1(x)
+
+            expect = linear_interp_np(
+                x_data, out_w=64, align_mode=1, align_corners=False
+            )
+
+            np.testing.assert_allclose(interp.numpy(), expect, rtol=1e-05)
+
+
 class TestLinearInterpOpFP16(TestLinearInterpOp):
     def test_check_output(self):
         self.check_output(atol=1e-3)