可选包的测试警告

Question

我的代码提供了使用可选包的可选功能。

当用户导入包含这些可选功能的包时，如果他/她没有安装附加包，则会显示警告。例如：

请安装optional-package 以使用所有功能

为了测试这些可选功能，可选包作为依赖安装在 tox 创建的 virtualenv 中：

# tox.ini
[my-env]
deps = optional-package

问题是我想测试一下：

1. 当用户未安装软件包时会显示警告
2. 当用户有包时不显示警告

但是，python 似乎没有提供卸载包的方法。

如果同时我需要tox 将其作为依赖项自动安装，我如何创建一个测试来描述用户在其系统上没有可选包的上下文？

Answer 1

这是测试some_module 在optional.package 不可用时引发警告的一种方法：

import some_module

# ...

    @mock.patch.dict("sys.modules", {"optional.package": None})
    def test_warning_optional_package_not_installed(self):
        with self.assertWarns(SomeWarning):
            imp.reload(some_module)

或者，您可以在 tox.ini 文件中定义多个测试环境，包括或不包括可选包。你可以配置 tox 设置环境变量，具体取决于是否安装了包：

[tox]
envlist = {with,without}_optional

[testenv]
basepython = python3.5
deps =
    required-package
    with_optional: optional-package
setenv = 
    with_optional: OPTIONAL_PACKAGE_AVAILABLE = true
commands = python -m unittest tests

然后您可以根据环境变量的值启用和禁用测试：

@unittest.skipUnless(os.env["OPTIONAL_PACKAGE_AVAILABLE"] == "true")
def test_no_warning_optional_package(self):
    with warnings.catch_warnings(record=True) as w:
        imp.reload(some_module)
    self.assertFalse(w)