Luigi 参数默认值和模拟

Question

我正在尝试模拟为 luigi 参数提供默认值的东西。

一个愚蠢的例子展示了我想要完成的事情：

正在测试的任务：

import luigi
from bar import Bar

bar = Bar()

class Baz(luigi.Task):

    qux = luigi.Parameter(default=bar.bar())

    def baz(self):
        return self.qux;

    def foo(self):
        return bar.bar()

单元测试代码：

import unittest
from mock import Mock, patch
from sut.baz import Baz

class TestMocking(unittest.TestCase):

    def test_baz_bar(self):
        self.assertEquals("bar", Baz().baz())

    @patch('sut.baz.bar')
    def test_patched_baz(self, mock_bar):
        mock_bar.bar = Mock(return_value="foo")
        self.assertEquals("foo", (Baz().baz()))

    @patch('sut.baz.bar')
    def test_patched_foo(self, mock_bar):
        mock_bar.bar = Mock(return_value="foo")
        self.assertEquals("foo", (Baz().foo()))

luigi.Parameter 逻辑似乎比补丁更早发生。

在本例中，test_patched_foo 通过，test_patched_baz 失败。所以补丁确实发生了，但发生在来自luigi.Parameter(default=bar.bar()) 行的调用之后。

是否可以模拟和修补以这种方式调用的东西？

Answer 1

尝试将qux = luigi.Parameter(default=bar.bar()) 行移动到Baz 类的__init__ 方法中。将它放在__init__ 之外，它是在类定义时设置的，而不是实例创建，但是将它放入__init__ 将延迟它的创建到Baz 实例创建的位置。不要忘记在super 类上调用__init__：

class Baz(luigi.Task):

    def __init__(self, *args, **kwargs):
        super(Baz, self).__init__(*args, **kwargs)

        self.qux = luigi.Parameter(default=bar.bar())

    ...