如何为依赖导入模块的代码编写单元测试

Question

拥有以下内容的文件

import something

if something:
   # do some things here
else:
    raise Exception

如何编写涵盖上述文件else分支的单元测试？

Answer 1

因此，您希望在测试中“取消导入”模块。您可以通过以下方式从命名空间中删除引用：

del something

是的，它不会进入else块。然后你可以用try-except块重写主模块，如下所示。

try: 
    if something: 
        print "Yes" 
except NameError: 
    print "Something not imported"