如何嘲笑/一個列表

Question

基本上我想這樣做替代類：

class SeaLion(object): 
    def __iter__(self): 
     return self 
    def next(self): 
     ... 

def train(): 
    sea_lion = SeaLion(...) 
    for tooth in sea_lion: 
     .. do something 

然後在我的測試文件我有：

class TestTrain(TestCase): 
    @patch('module.SeaLion') 
    def test_train(self, mock_SeaLion): 
     mock_SeaLion.return_value = [1,2,3] 
     ... 

基本上我想sea_lion在訓練函數是一個列表而不是該類的一個實例。

與上面的代碼sea_lion是MagicMock（名稱= '海獅（）'）

Answer 1

嘗試這種情況：

import unittest 
from mock import patch 

class SeaLion(object): 
    def __iter__(self): 
     return self 

    def next(self): 
     pass 

def train(): 
    sea_lion = SeaLion() 
    print(repr(sea_lion)) 
    for tooth in sea_lion: 
     return True 

    return False 

class TestTrain(unittest.TestCase): 
    @patch('__main__.SeaLion') 
    def test_train(self, mock_SeaLion): 
     mock_SeaLion.return_value = [1] 
     self.assertTrue(train()) 

if __name__ == '__main__': 
    unittest.main() 

結果：

[1] 
. 
---------------------------------------------------------------------- 
Ran 1 test in 0.044s 

OK