nosetest一個隨機元素的函數

Question

嗨我是新的單元測試，想知道如何測試一個函數，它有一個隨機組件？

我有以下Python代碼：

class Questions(object): 

def __init__(self): 

    self.questions = {} 

place_value = { 
    0: "Thousands", 
    1: "Hundreads", 
    2: "Tens", 
    3: "Units/ones", 
} 

def place_value(self, question, number): 

    selection = randint(0, 3) 

    number = ''.join(map(str, number)) 
    value = number[selection] 

    question_text = question.format(value, number) 

    li = generate_a_list(question_text) 

    self.questions['question1'] = li 

測試代碼

def test_place_value(): 

    obj = math_q.Questions() 
    obj.place_value("value of {0} in {1}", [1,2,3,4]) 

    assert_equal(obj.questions["question1"], ["value of {0} in 1234"]) 

的問題是不知道哪個值1-4從「值=數選擇了[選擇]'的代碼。

可以做些什麼呢？ 謝謝。

Answer 1

假設你的代碼是path/my_file.py你可以使用模擬模塊是這樣的：

@mock.patch('path.my_file.randint', return_value=0) 
def test_place_value(m_randint): 
    obj = math_q.Questions() 
    obj.place_value("value of {0} in {1}", [1,2,3,4]) 

    m_randint.assert_called_once_with(0, 3) 
    assert obj.questions["question1"] == ... 

然後只寫4個測試來處理所有案件。

附註：我強烈建議切換到pytest。