diff --git a/python/paddle/fluid/tests/unittests/test_batch_norm_op.py b/python/paddle/fluid/tests/unittests/test_batch_norm_op.py index 5e3c41441dbb12078cbd203c05ed35dd0ee8e49a..851dfe39e48ea16699b3baa5ad39cc33c32dfb03 100644 --- a/python/paddle/fluid/tests/unittests/test_batch_norm_op.py +++ b/python/paddle/fluid/tests/unittests/test_batch_norm_op.py @@ -575,7 +575,7 @@ class TestBatchNormOpFreezeStatsAndScaleBiasTraining( self.fetch_list = ['y', 'mean', 'variance', 'x@GRAD'] -class TestBatchNormOpError(OpTest): +class TestBatchNormOpError(unittest.TestCase): def test_errors(self): with program_guard(Program(), Program()): # the input of batch_norm must be Variable. diff --git a/python/paddle/fluid/tests/unittests/test_expand_as_op.py b/python/paddle/fluid/tests/unittests/test_expand_as_op.py index 836e49c54e2dac344761c1717dfad7724a23f3dd..c5d09ac3cd19a89e5c7c4d0a56efd308549648b4 100755 --- a/python/paddle/fluid/tests/unittests/test_expand_as_op.py +++ b/python/paddle/fluid/tests/unittests/test_expand_as_op.py @@ -103,7 +103,7 @@ class TestExpandAsOpRank4(OpTest): # Test python API -class TestExpandAPI(OpTest): +class TestExpandAPI(unittest.TestCase): def test_api(self): input1 = np.random.random([12, 14]).astype("float32") input2 = np.random.random([48, 14]).astype("float32") diff --git a/python/paddle/fluid/tests/unittests/test_gather_nd_op.py b/python/paddle/fluid/tests/unittests/test_gather_nd_op.py index 357b2dc060742e545ce896b60373fe39badf4e3d..724b8040354d5f713d163b699e2884161bce48b9 100644 --- a/python/paddle/fluid/tests/unittests/test_gather_nd_op.py +++ b/python/paddle/fluid/tests/unittests/test_gather_nd_op.py @@ -128,7 +128,7 @@ class TestGatherNdOpWithHighRankDiff(OpTest): #Test Python API -class TestGatherNdOpAPI(OpTest): +class TestGatherNdOpAPI(unittest.TestCase): def test_case1(self): x1 = fluid.layers.data( name='x1', shape=[30, 40, 50, 60], dtype='float32') @@ -147,7 +147,7 @@ class TestGatherNdOpAPI(OpTest): #Test Raise Index Error -class TestGatherNdOpRaise(OpTest): +class TestGatherNdOpRaise(unittest.TestCase): def test_check_raise(self): def check_raise_is_test(): try: diff --git a/python/paddle/fluid/tests/unittests/test_locality_aware_nms_op.py b/python/paddle/fluid/tests/unittests/test_locality_aware_nms_op.py index bed185551b77f3e3bc103cc0dbfd4adceb0f78c5..e4845c90a60bf665363195a2766a5cdda9847203 100644 --- a/python/paddle/fluid/tests/unittests/test_locality_aware_nms_op.py +++ b/python/paddle/fluid/tests/unittests/test_locality_aware_nms_op.py @@ -307,7 +307,7 @@ class TestLocalAwareNMSOp4Points(OpTest): self.check_output() -class TestLocalityAwareNMSAPI(OpTest): +class TestLocalityAwareNMSAPI(unittest.TestCase): def test_api(self): boxes = fluid.data(name='bboxes', shape=[None, 81, 8], dtype='float32') scores = fluid.data(name='scores', shape=[None, 1, 81], dtype='float32') diff --git a/python/paddle/fluid/tests/unittests/test_lrn_op.py b/python/paddle/fluid/tests/unittests/test_lrn_op.py index 7ec329d9f6c224e60a8133abb654207ea5c12266..bf6b555c4663781bcad50b282bb6d38969521520 100644 --- a/python/paddle/fluid/tests/unittests/test_lrn_op.py +++ b/python/paddle/fluid/tests/unittests/test_lrn_op.py @@ -106,7 +106,7 @@ class TestLRNOpAttrDataFormat(TestLRNOp): self.data_format = 'NHWC' -class TestLRNAPI(OpTest): +class TestLRNAPI(unittest.TestCase): def test_case(self): data1 = fluid.data(name='data1', shape=[2, 4, 5, 5], dtype='float32') data2 = fluid.data(name='data2', shape=[2, 5, 5, 4], dtype='float32') diff --git a/python/paddle/fluid/tests/unittests/test_nearest_interp_op.py b/python/paddle/fluid/tests/unittests/test_nearest_interp_op.py index a8d16677e22127a484e1c6987d7f6d47fdc220bd..bb90279ea938024e2bc342351782ef932b49193c 100755 --- a/python/paddle/fluid/tests/unittests/test_nearest_interp_op.py +++ b/python/paddle/fluid/tests/unittests/test_nearest_interp_op.py @@ -423,7 +423,7 @@ class TestNearestInterp_attr_tensor_Case3(TestNearestInterpOp_attr_tensor): self.scale_by_1Dtensor = True -class TestNearestAPI(OpTest): +class TestNearestAPI(unittest.TestCase): def test_case(self): x = fluid.data(name="x", shape=[2, 3, 6, 6], dtype="float32") y = fluid.data(name="y", shape=[2, 6, 6, 3], dtype="float32") @@ -474,7 +474,7 @@ class TestNearestAPI(OpTest): self.assertTrue(np.allclose(results[i + 1], expect_res)) -class TestNearestInterpException(OpTest): +class TestNearestInterpException(unittest.TestCase): def test_exception(self): input = fluid.data(name="input", shape=[1, 3, 6, 6], dtype="float32") diff --git a/python/paddle/fluid/tests/unittests/test_print_op.py b/python/paddle/fluid/tests/unittests/test_print_op.py index 49b55e376b41811be37c7f149565480a45b556da..21bf3f0b2870c573d5e0ee576a3c829be5c33d71 100644 --- a/python/paddle/fluid/tests/unittests/test_print_op.py +++ b/python/paddle/fluid/tests/unittests/test_print_op.py @@ -82,7 +82,7 @@ class TestPrintOpCPU(unittest.TestCase): return_numpy=False) -class TestPrintOpError(OpTest): +class TestPrintOpError(unittest.TestCase): def test_errors(self): with program_guard(Program(), Program()): # The input type of Print_op must be Variable. diff --git a/python/paddle/fluid/tests/unittests/test_unsqueeze2_op.py b/python/paddle/fluid/tests/unittests/test_unsqueeze2_op.py index 0e553287b568ecc5477929122bc27ff35f1a2882..51fb7302960f574b4c1d09a8199797a636842666 100644 --- a/python/paddle/fluid/tests/unittests/test_unsqueeze2_op.py +++ b/python/paddle/fluid/tests/unittests/test_unsqueeze2_op.py @@ -203,7 +203,7 @@ class TestUnsqueezeOp4_AxesTensor(TestUnsqueezeOp_AxesTensor): # test api -class TestUnsqueezeAPI(OpTest): +class TestUnsqueezeAPI(unittest.TestCase): def test_api(self): input = np.random.random([3, 2, 5]).astype("float32") x = fluid.data(name='x', shape=[3, 2, 5], dtype="float32") diff --git a/python/paddle/fluid/tests/unittests/test_zeros_op.py b/python/paddle/fluid/tests/unittests/test_zeros_op.py index 1be2f4d04791b76be014e34e1752caab43157fc7..6ce39cd63767e3dac474a7b0b57bfec1438d5e09 100644 --- a/python/paddle/fluid/tests/unittests/test_zeros_op.py +++ b/python/paddle/fluid/tests/unittests/test_zeros_op.py @@ -24,7 +24,7 @@ import paddle.fluid as fluid from paddle.fluid import compiler, Program, program_guard -class TestZerosOpError(OpTest): +class TestZerosOpError(unittest.TestCase): def test_errors(self): with program_guard(Program(), Program()): # The input dtype of zeros_op must be bool, float16, float32, float64, int32, int64.