如何為依賴導入模塊的代碼編寫單元測試

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"