如何使用assert_raises趕上SystemExit例外

Question

我有我想測試代碼：

from random import randint 

class End(object): 
      def __init__(self): 
      self.quips=['You dead', 'You broke everything you can','You turn you head off'] 

      def play(self): 
       print self.quips[randint(0, len(self.quips)-1)] 
       sys.exit(1) 

如何檢查是否退出或者是不使用accept_raises？

def test_End(): 
    end=End().play() 
    assert_raises(what should I put here) 

Answer 1

可以在assertRaises趕上SystemExit例外：

with self.assertRaises(SystemExit): 
    End().play() 

或修補sys.exit通過mock：

with patch.object(sys, 'exit') as mock_method: 
    End().play() 
    self.assertTrue(mock_method.called) 

Answer 2

我喜歡@raises裝飾

from nose.tools import raises 

@raises(SystemExit) 
def test_End(): 
    end=End().play()