class TestTransformProcessTask(TestCase): def __init__(self, methodName='runTest'): super(TestTransformProcessTask, self).__init__(methodName) category_transformer = lambda row: 'category.{0!s}'.format(row[0]) data_transformer = lambda row: '{0!s}:{1!s}'.format(row[1], row[2]) self._process_task = TransformProcessTask(category_transformer, data_transformer) def test_process_empty_list(self): self.assertEqual([], self._process_task.process([])) def test_process_list_with_correct_tuples(self): expected = [DataItem('category.cat1', 'data1:subdata1'), DataItem('category.cat2', 'data2:subdata2'), DataItem('category.cat1', 'data3:subdata3'), DataItem('category.cat3', 'data4:subdata4')] source = [[('cat1', 'data1', 'subdata1'), ('cat2', 'data2', 'subdata2')], [('cat1', 'data3', 'subdata3')], [('cat3', 'data4', 'subdata4')]] self.assertEqual(expected, self._process_task.process(source)) def test_process_list_with_incorrect_tuples(self): source = [[('cat1', 'data1')]] self.assertRaises(IndexError, lambda: self._process_task.process(source))
def __init__(self, methodName='runTest'): super(TestTransformProcessTask, self).__init__(methodName) category_transformer = lambda row: 'category.{0!s}'.format(row[0]) data_transformer = lambda row: '{0!s}:{1!s}'.format(row[1], row[2]) self._process_task = TransformProcessTask(category_transformer, data_transformer)