python - 如何修复 PytestAssertRewriteWarning？

Question

我刚刚开始使用 pytest 和 pytest_mysql 并拥有以下代码:

import pytest
from pytest_mysql import factories

if __name__ == "__main__":
    pytest.main([r'test.py', '-v'])
    testdb = factories.mysql('testdb')

这会生成警告:

PytestAssertRewriteWarning: Module already imported so cannot be rewritten: pytest_mysql

1. 我应该担心吗？
2. 如何确保我不会收到警告？

Answer 1

这就是我解决它的方法!

import pytest
if __name__=='__main__': #Note 1
    pytest.main()
#
#Your code goes here
#
from pytest_mysql import factories
testdb = factories.mysql('testdb')

因此，在导入其他内容之前尝试运行 pytest.main() ..

注1

这不是您通常调用__name__=='__main__'的方式，但因为它有效，所以我买了它!