python - 如何通过字典进行测试
问题描述
我在编写测试(对于python)方面很新,所以我现在有一个问题,我怎样才能将字典传递给测试函数?目前,我执行以下操作:
import os
import sys
import shutil
from app.views import file_io
import pytest
from tempfile import mkdtemp
import codecs
@pytest.fixture()
def tempdir():
tempdir = mkdtemp()
yield tempdir
shutil.rmtree(tempdir)
articles = [
["", "README.md", "# Hallo Welt", "<h1>Hallo Welt</h1>\n"],
["test", "article.md", "# Hallo Welt", "<h1>Hallo Welt</h1>\n"]
]
@pytest.mark.parametrize("dir, file, content_plain, content_md", articles)
def test_readRaw(tempdir, dir, file, content_plain, content_md):
dest_path=os.path.join(tempdir, dir)
os.makedirs(dest_path, exist_ok=True)
with codecs.open(os.path.join(dest_path, file), 'w', 'utf-8') as fh:
fh.write(content_plain)
assert file_io.readRaw(os.path.join(dest_path, file)) == content_plain
我的想法/希望是我可以修改代码,以便可以执行以下操作:
articles = [
{ "dir": "",
"filename": "README.md",
"content_md": "# Hello World",
"content_html": "<h1>Hello World</h1>\n" },
{ "dir": "test",
"filename": "article.md",
"content_md": "# Hallo Welt",
"content_html": "<h1>Hallo Welt</h1>\n"}
]
@pytest.mark.parametrize(**articles, articles)
def test_readRaw(tempdir, **articles):
with codecs.open(os.path.join(dir, file), 'w', 'utf-8') as fh:
fh.write(content_md)
assert file_io.readRaw(os.path.join(dir, file)) == content_md
特别是我想避免提及所有键,所以如果我错过了一些东西而不修改所有测试,我可以扩展字典。
也许这是一个愚蠢的问题,但正如我所说,我只是从这个话题开始,所以我非常感谢每一个提示我该如何做到这一点(或者什么是更好的方法)。最好的问候丹
解决方案
不要尝试 splat / unsplat,而是尝试将article
其作为参数:
@pytest.mark.parametrize('article', articles)
def test_readRaw(tempdir, article):
# use `article['foo']` here...
另一种选择(利用 python3.6+ 功能)是手动扩展键 - 尽管您必须小心以相同的顺序定义每个字典
@pytest.mark.parametrize(tuple(articles[0]), [tuple(dct.values()) for dct in articles])
def test_readRaw(tempdir, dir, file, content_plain, content_md):
...
对于它的价值,我认为您会通过采用第二种方法来牺牲一些可读性(并使测试特别脆弱)
~相关建议
- 您可以使用内置的
tmp_path
/tmpdir
固定装置而不是构建自己的 - 你的被测函数实际上并不依赖于
dir
orfile
,你最好不要参数化这些
考虑到这两个因素,使用经典参数化(一个简单的输入/输出表),您的测试变得更加简单:
@pytest.mark.parametrize(
('content_plain', 'content_md'),
(
("# Hallo Welt", "<h1>Hallo Welt</h1>\n"),
("# ohai", "<h1>ohai</h1>\n"),
),
)
def test_readRaw(tmpdir, content_plain, content_md):
f = tmpdir.join('f')
f.write(content_plain)
assert file_io.readRaw(f) == content_md
免责声明:我是目前的核心开发者之一pytest
推荐阅读
- java - No validator could be found for constraint 'javax.validation.constraints.Size'
- r - dplyr - consecutive occurrences in same column, label depending on # of occurrences
- python - 后台进程和 tkinter
- python - 注释堆叠的 barplot matplotlib 和 pandas
- python - Why does `flask_bootstrap` fail to import?
- r - R 的官员包无法识别主布局中的 Powerpoint 占位符
- swift - How to encode SecKey to base64 string using swift
- html - JQuery 自动完成永远不会触发
- opencv - Is it possible to find camera position using 8-10 non-coplanar points, if their 3D coordinates are unknown?
- vue.js - 在 Vue.js 模板的每个循环中显示 JSON 数据属性