在pytest中满足条件时只处理一部分测试用例

Question

我有以下 pytest 案例，我需要一个条件或异常，如果类别是蓝色，则应创建所有 4(P1 - P4)。

如果它是灰色的，那么应该只创建 P1 和 P2。

但是，测试用例应该在两个类别中都通过，我需要在同一个测试用例中实现两者。

我是新手，所以我需要帮助。 下面的代码片段中提供了更多详细信息。

....
....
category = eval(data_cim['generic']['category']) 

类别是蓝色或灰色。 蓝色类别有所有 P1 - P4，灰色只有 P1 和 P2

def test_01(self):
        resp = self.x.create_p(self.category.p1)
        resp_status = self.x.checkResponseStatus(resp)
        msg = self.x.extraceMessage(resp.text)
        info = msg['error']
        self.assertEqual(resp_status, RESTResponse.PASSED, 'Creating P1 failed')
        self.assertEqual('OK',info, 'Creating P1 failed')

        resp = self.x.create_p(self.category.p2)
        resp_status = self.x.checkResponseStatus(resp)
        msg = self.x.extraceMessage(resp.text)
        info = msg['error']
        self.assertEqual(resp_status, RESTResponse.PASSED, 'Creating P2 failed')
        self.assertEqual('OK',info, 'Creating P2 failed')

如果类别 = 蓝色，继续下 2 个创建如果类别为灰色，则停止执行测试用例，因为灰色只有 p1 和 p2

        resp = self.x.create_p(self.category.p3)
        resp_status = self.x.checkResponseStatus(resp)
        msg = self.x.extraceMessage(resp.text)
        info = msg['error']
        self.assertEqual(resp_status, RESTResponse.PASSED, 'Creating P3 failed')
        self.assertEqual('OK',info, 'Creating P3 failed')

        resp = self.x.create_p(self.category.p4)
        resp_status = self.x.checkResponseStatus(resp)
        msg = self.x.extraceMessage(resp.text)
        info = msg['error']
        self.assertEqual(resp_status, RESTResponse.PASSED, 'Creating P4 failed')
        self.assertEqual('OK',info, 'Creating P4 failed')