コード例 #1
0
ファイル: mock_test.py プロジェクト: srirampmalladi/luigi
    def test_with(self):
        t = MockFile("foo")
        with t.open('w') as b:
            b.write("bar")

        with t.open('r') as b:
            self.assertEqual(list(b), ['bar'])
コード例 #2
0
ファイル: mock_test.py プロジェクト: rudimk/luigi
    def test_unicode(self):
        t = MockFile("foo")
        with t.open('w') as b:
            b.write(u"bar")

        with t.open('r') as b:
            self.assertEqual(b.read(), u'bar')
コード例 #3
0
ファイル: mock_test.py プロジェクト: NeverC/luigi
    def test_with(self):
        t = MockFile("foo")
        with t.open('w') as b:
            b.write("bar")

        with t.open('r') as b:
            self.assertEquals(list(b), ['bar'])
コード例 #4
0
ファイル: range_test.py プロジェクト: rudimk/luigi
 def output(self):
     base = self.dh.strftime('/even/%Y%m%d%H')
     if self.dh.hour % 2 == 0:
         return MockFile(base)
     else:
         return {
             'spi': MockFile(base + '/something.spi'),
             'spl': MockFile(base + '/something.spl'),
         }
コード例 #5
0
ファイル: mock_test.py プロジェクト: g152xx/luigi
    def test_1(self):
        t = MockFile("test")
        p = t.open("w")
        print >> p, "test"
        p.close()

        q = t.open("r")
        self.assertEqual(list(q), ["test\n"])
        q.close()
コード例 #6
0
ファイル: mock_test.py プロジェクト: srirampmalladi/luigi
    def test_1(self):
        t = MockFile('test')
        p = t.open('w')
        print >> p, 'test'
        p.close()

        q = t.open('r')
        self.assertEqual(list(q), ['test\n'])
        q.close()
コード例 #7
0
 def test_mockfile(self):
     from luigi.mock import MockFile
     if six.PY3:
         with self.assertWarnsRegex(
                 DeprecationWarning,
                 r'MockFile has been renamed MockTarget'):
             self.assertTrue(isinstance(MockFile('foo'), MockTarget))
     else:
         self.assertTrue(isinstance(MockFile('foo'), MockTarget))
コード例 #8
0
ファイル: mock_test.py プロジェクト: NeverC/luigi
    def test_1(self):
        t = MockFile('test')
        p = t.open('w')
        print >> p, 'test'
        p.close()

        q = t.open('r')
        self.assertEqual(list(q), ['test\n'])
        q.close()
コード例 #9
0
    def output(self):
        """
        Returns the target output for this task.
        In this case, a successful execution of this task will simulate the creation of a file in a filesystem.

        :return: the target output for this task.
        :rtype: object (:py:class:`~luigi.target.Target`)
        """
        return MockFile("output", mirror_on_stderr=True)
コード例 #10
0
ファイル: range_test.py プロジェクト: leochencipher/luigi
 def test_missing_tasks_correctly_required(self):
     for task_path in task_a_paths:
         MockFile(task_path)
     task = RangeHourly(now=datetime_to_epoch(datetime.datetime(2016, 4, 1)),
                        of='TaskA',
                        start=datetime.datetime(2014, 3, 20, 17),
                        task_limit=3,
                        hours_back=3 * 365 * 24)  # this test takes a few seconds. Since stop is not defined, finite_datetimes constitute many years to consider
     actual = [t.task_id for t in task.requires()]
     self.assertEqual(actual, expected_a)
コード例 #11
0
class MockObject(Target):

    mock_file = MockFile('whatever')

    def exists(self):
        return True

    def write(self, obj):
        with self.mock_file.open('w') as _out:
            _out.write(pickle.dumps(obj))

    def read(self):
        with self.mock_file.open('r') as _in:
            return pickle.reads(_in.read())
コード例 #12
0
 def output(self):
     return MockFile(self.date.strftime('/tmp/copy-data-%Y-%m-%d.txt'))
コード例 #13
0
ファイル: test_event_callbacks.py プロジェクト: rudimk/luigi
 def output(self):
     return MockFile('/%s/%u' % (self.__class__.__name__, self.param))
コード例 #14
0
ファイル: mock_test.py プロジェクト: NeverC/luigi
 def _touch(self, path):
     t = MockFile(path)
     with t.open('w'):
         pass
コード例 #15
0
ファイル: luigi_pipeline.py プロジェクト: gaarangoa/nanoARG
 def output(self):
     return MockFile("done.txt")
コード例 #16
0
 def output(self):
     return MockFile(
         self.dh.strftime('TaskB/%%s%Y-%m-%d/%H') % self.complicator)
コード例 #17
0
ファイル: range_test.py プロジェクト: rudimk/luigi
 def output(self):
     return MockFile(
         self.d.strftime(
             '/data/2014/p/v/z/%Y_/_%m-_-%doctor/20/ZOOO'))
コード例 #18
0
 def output(self):
     return MockFile("output", mirror_on_stderr=True)
コード例 #19
0
 def output(self):
     return MockFile("in-memory-paths")
コード例 #20
0
ファイル: range_test.py プロジェクト: leochencipher/luigi
 def output(self):
     return MockFile(self.dh.strftime('not/a/real/path/%Y-%m-%d/%H'))
コード例 #21
0
 def output(self):
     return MockFile(
         self.d.strftime('/n2000y01a05n/%Y_%m-_-%daww/21mm01dara21/ooo'))
コード例 #22
0
 def output(self):
     return MockFile(self.dh.strftime('TaskA/%Y-%m-%d/%H'))
コード例 #23
0
 def output(self):
     return MockFile("MockInput")
コード例 #24
0
ファイル: luigi_pipeline.py プロジェクト: gaarangoa/nanoARG
 def output(self):
     parameters.getParameters(self.sid, remote_storage_root)
     return MockFile("output", mirror_on_stderr=True)
コード例 #25
0
 def output(self):
     return MockFile('whatever')
コード例 #26
0
ファイル: mock_test.py プロジェクト: srirampmalladi/luigi
 def _touch(self, path):
     t = MockFile(path)
     with t.open('w'):
         pass
コード例 #27
0
 def test_mockfile(self):
     from luigi.mock import MockFile
     self.assertTrue(isinstance(MockFile('foo'), MockTarget))
コード例 #28
0
ファイル: sqla_test.py プロジェクト: leochencipher/luigi
 def output(self):
     return MockFile("BaseTask", mirror_on_stderr=True)
コード例 #29
0
ファイル: parameter_test.py プロジェクト: su-admin/luigi
 def output(self):
     return MockFile('banana-%s-%s' % (self.x, self.y))
コード例 #30
0
 def output(self):
     return MockFile("MyOutput1")
コード例 #31
0
 def output(self):
     return MockFile("AddTaxiLocation")