蟒/ django的單元測試功能倍率

Question

我與非預定數量內它的迭代耗時的方法，我需要測試：

def testmethod(): 
    objects = get_objects() 
    for objects in objects: 
     # Time-consuming iterations 
     do_something(object) 

一次迭代是足以測試我。僅使用一次迭代測試此方法的最佳做法是什麼？

Answer 1

也許把你方法引入

def my_method(self, objs=None): 
    if objs is None: 
     objs = get_objects() 
    for obj in objs: 
     do_something(obj) 

然後在您的測試，你可以用自定義的OBJ參數調用它。

Answer 2

更新：

我誤解了原來的問題，所以這裏的我會怎樣解決這個問題，而不改變源代碼：

LAMBDA你的調用來獲取對象返回一個對象。例如：

from your_module import get_objects 

def test_testmethdod(self): 
    original_get_objects = get_objects 
    one_object = YourObject() 
    get_objects = lambda : [one_object,] 

    # ... 
    # your tests here 
    # ... 

    # Reset the original function, so it doesn't mess up other tests 
    get_objects = original_get_objects