fix bug

PaddlePaddle · Aug 29, 2022 · 72265b4 · 72265b4
1 parent d5b45e4
commit 72265b4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/paddle/fluid/tests/unittests/test_elementwise_mod_op.py b/python/paddle/fluid/tests/unittests/test_elementwise_mod_op.py
@@ -136,13 +136,13 @@ def test_dygraph(self):
             np.testing.assert_allclose(z_expected, z.numpy(), rtol=1e-05)
 
 
-class TestRemainderInplaceOp(unittest.TestCase):
+class TestRemainderInplaceOp(TestRemainderOp):
 
     def _executed_api(self, x, y, name=None):
         return x.remainder_(y, name)
 
 
-class TestRemainderInplaceBroadcastSuccess(unittest.TestCase):
+class TestRemainderInplaceBroadcastSuccess(TestRemainderOp):
 
     def init_data(self):
         self.x_numpy = np.random.rand(2, 3, 4).astype('float')