コード例 #1
0
ファイル: runner_test.py プロジェクト: don9z/pydoop
    def test_combine_bykeys(self):
        test_stdin = ['1\tv1\tv2\tv3',
                      '1\tv4\tv5\tv6',
                      '2\tv21\tv22\tv23',
                      '2\tv24\tv25\tv26',
                      '2\tv27\tv28\tv29',
                      '3\tv31\tv32\tv33']
        expected = {'1' : [['v1', 'v2', 'v3'],
                           ['v4', 'v5', 'v6']],
                    '2' : [['v21', 'v22', 'v23'],
                           ['v24', 'v25', 'v26'],
                           ['v27', 'v28', 'v29']],
                    '3' : [['v31', 'v32', 'v33']]}
        result = {}
        for keyvalues in runner.combine_bykeys(test_stdin):
            result[keyvalues[0]] = keyvalues[1]

        self.assertEqual(expected, result)
コード例 #2
0
ファイル: runner_test.py プロジェクト: don9z/pydoop
 def test_combine_bykeys_when_empty_input(self):
     test_stdin = []
     for keyvalues in runner.combine_bykeys(test_stdin):
         # Should emit nothing
         self.assertIsNone(1)