我有一个内部使用lambda表达式进行处理的函数。我需要为这个函数编写单元测试。当我使用命令运行我的python单元测试时: python -m unittest discover -v,单元测试无限等待回调函数(在本例中为lambda表达式)。
我们可以对这些案件进行单元测试吗
示例代码(abc.py)
def uses_callback_func():
super_sf= Sframe()
.... # some other code for sframe creation.
super_sf['weighted_innsa'] = super_sf.apply(lambda x:
2.0 if (x['from_id'] =='INNSA' or x['to_id'] == 'INNSA') # unittest halts at this line.
else 1.0)
return len(super_sf['weighted_innsa']
test_abc.py
def test_callback_func(self):
value_1 = abc.uses_callback_func()
assertEqual(value_1, 3)