如何测试功能取决于python中的另一个功能

Question

我需要测试以下类型的代码：

list = [1,2,3,4]

def getData(list):
    return list[0] + list[1]

def processData():
    data = getData(list)    
    multiply = data*data
    return multiply

def test_functions():
    assert getData([0,1]) == 1
    assert processData() == 1

如何告诉测试我需要数据= getData（[0,1]），所以基本上用测试值替换数据。