Python EjudgeRunsFilesWorkerの例

コード例 #1

ファイル: walker_test.py プロジェクト: parallel-p/pesto

class TestEjudgeRunsFilesWorker(PestoTestCase):
    def setUp(self):
        self.worker = EjudgeRunsFilesWorker()
        self.os_walk_backup = os.walk

    def tearDown(self):
        os.walk = self.os_walk_backup

    def test_common(self):
        os.walk = Mock(return_value=[('a', ['b'], ['c', 'f.gz']), ('b', ['c'], ['d'])])
        self.assertEqual(list(self.worker.walk('a')), [('xml', os.path.join('a', 'c')),
                                                       ('gzip', os.path.join('a', 'f.gz')),
                                                       ('xml', os.path.join('b', 'd'))])

    def test_empty(self):
        os.walk = Mock(return_value=[])
        self.assertEqual(list(self.worker.walk('')), [])

コード例 #2

ファイル: walker_test.py プロジェクト: parallel-p/pesto

 def setUp(self):
     self.worker = EjudgeRunsFilesWorker()
     self.os_walk_backup = os.walk