使用树状数据进行 pytest 嵌套参数化

使用 pytest，我尝试测试像树一样的分层场景。 让我们以文档结构为例：

Document --- Chapter --- Paragraph
       1   n        1   n

当文档包含多个章节时；一章包含多个段落。

当开始测试新文档时，需要运行一些设置代码；当新的章节开始时，需要运行一些其他设置代码；段落也一样。

写成伪代码：

for doc in documents:
    setup_doc(doc)
    for chapter in doc.chapters:
        setup_chapter(chapter)
        for paragraph in chapter.paragraphs:
            setup_paragraph(paragraph)
            test_it(doc, chapter, paragraph)
            teardown_paragraph(paragraph)
        teardown_chapter(chapter)
    teardown_doc(doc)

如果我们有以下数据：

Document Alpha
   chapter A
      Paragraph A1
      Paragraph A2
   chapter B
      Paragraph B1

我希望收集的测试用例是：

test_it[Alpha, A, A1]
test_it[Alpha, A, A2]
test_it[Alpha, B, B1]

我尝试了 pytest_generate_tests、类场景、固定装置和参数化测试函数的不同组合，但未能实现这一点。

任何指示将不胜感激。

Pytest 装置应该是独立的。 因此，为了解决您的任务，您必须构建一个带有所有功能组合的简单列表（文档 - 章节 - 段落）的固定装置。

您可以使用返回此类列表的一个元素的简单固定装置来完成此操作，也可以在测试生成阶段生成此列表，如下面的代码所示。

documents = {
        'Alpha': {
            'A': {'A1': {},'A2': {}},
            'B': {'B1': {}}
        }
    }


def pytest_generate_tests(metafunc):
    """Documents tree from documents"""
    if 'document' in metafunc.fixturenames:
        documents_plain = []
        for document in documents.keys():
            for chapter in documents[document].keys():
                for paragraph in documents[document][chapter].keys():
                    documents_plain.append({'document': document, 'chapter': chapter, 'paragraph': paragraph})
        metafunc.parametrize(
            'document',
            documents_plain,
            scope='session')


def test_it(document):
    print('{}, {}, {}'.format(document['document'], document['chapter'], document['paragraph']))


py.test -s

Alpha, B, B1
Alpha, A, A1
Alpha, A, A2