0
点赞
收藏
分享

微信扫一扫

人工智能学习笔记 - 初级篇Ⅱ - 图形可视化 - 第10节: 绘制饼图

一个会话级别的夹具(fixture)可以有效地访问所有收集到的测试项。以下是一个夹具函数的示例,该函数遍历所有收集到的测试项,检查它们的测试类是否定义了一个名为 callme 的方法,并调用它:

# conftest.py 文件的内容  

import pytest

@pytest.fixture(scope="session", autouse=True)
def callattr_ahead_of_alltests(request):
print("callattr_ahead_of_alltests called")
seen = {None} # 用于跟踪已经检查过的类,初始化为包含 None 的集合
session = request.node # 获取当前会话的根节点
for item in session.items: # 遍历会话中的所有测试项
cls = item.getparent(pytest.Class) # 获取测试项所属的测试类(如果存在的话)
if cls not in seen: # 如果这个类还没有被检查过
if hasattr(cls.obj, "callme"): # 检查这个类的实例(cls.obj)是否有 `callme` 方法
cls.obj.callme() # 如果有,就调用它
seen.add(cls) # 将这个类添加到已检查的集合中

现在,测试类可以定义一个 callme 方法,该方法将在运行任何测试之前被调用:

# test_module.py 文件的内容  

class TestHello:
@classmethod
def callme(cls):
print("callme called!")

def test_method1(self):
print("test_method1 called")

def test_method2(self):
print("test_method2 called")


class TestOther:
@classmethod
def callme(cls):
print("callme other called")

def test_other(self):
print("test other")

# works with unittest as well ...
import unittest


class SomeTest(unittest.TestCase):
@classmethod
def callme(self):
print("SomeTest callme called")

def test_unit1(self):
print("test_unit1 method called")

如果你在没有捕获输出的情况下运行这个命令:

$ pytest -q -s test_module.py

你将看到以下输出:

callattr_ahead_of_alltests called  
callme called!
callme other called
SomeTest callme called
test_method1 called
.test_method2 called
.test other
.test_unit1 method called
.
4 passed in 0.12s
举报

相关推荐

0 条评论