Unittest一个使用lambda表达式的函数

时间:2016-07-15 08:07:22

标签: python-2.7 lambda python-unittest

我有一个内部使用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)  

0 个答案:

没有答案