def test_tasks(self):
t = Task(description = 'T1')
list = group_by([t], lambda t: t.importance())
self.assertEqual(list[0][0], t.importance())
self.assertEqual(list[0][1], [t])
self.assertEqual([t], flatten(list))
def test_two_equal(self):
gb = self.group([1,1])
self.assertEqual([[1, [1,1]]], gb)
self.assertEqual([1,1], flatten(gb))
def test_two_different(self):
gb = self.group([1,2])
self.assertEqual([[1, [1]], [2, [2]]], gb)
self.assertEqual([1,2], flatten(gb))
def get_tasks(response):
return flatten(response.context["tasks"])
