Python flattenの例

コード例 #1

ファイル: test_models.py プロジェクト: nodet/yata

 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))

コード例 #2

ファイル: test_models.py プロジェクト: nodet/yata

 def test_two_equal(self):
     gb = self.group([1,1])
     self.assertEqual([[1, [1,1]]], gb)
     self.assertEqual([1,1], flatten(gb))

コード例 #3

ファイル: test_models.py プロジェクト: nodet/yata

 def test_two_different(self):
     gb = self.group([1,2])
     self.assertEqual([[1, [1]], [2, [2]]], gb)
     self.assertEqual([1,2], flatten(gb))

コード例 #4

ファイル: test_views.py プロジェクト: nodet/yata

def get_tasks(response):
    return flatten(response.context["tasks"])