paulin/bonobo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[tests] rename factory test and move bag detecting so any bag is returned as is as an output.

Browse Source
This commit is contained in:
Romain Dorgueil
2017-10-23 21:28:49 +02:00
parent 9a54f7b4aa
commit ece764b95c
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
bonobo/execution/node.py
View File

@ -9,7 +9,7 @@ from bonobo.execution.base import LoopingExecutionContext
from bonobo.structs.bags import Bag from bonobo.structs.bags import Bag
from bonobo.structs.inputs import Input from bonobo.structs.inputs import Input
from bonobo.structs.tokens import Token from bonobo.structs.tokens import Token
from bonobo.util import get_name, iserrorbag, isloopbackbag from bonobo.util import get_name, iserrorbag, isloopbackbag, isbag
from bonobo.util.compat import deprecated_alias from bonobo.util.compat import deprecated_alias
from bonobo.util.statistics import WithStatistics from bonobo.util.statistics import WithStatistics
@ -142,7 +142,7 @@ def _resolve(input_bag, output):
if output is NOT_MODIFIED: if output is NOT_MODIFIED:
return input_bag return input_bag
if iserrorbag(output): if isbag(output):
return output return output
return Bag(output) return Bag(output)

2
tests/nodes/factory.py → tests/nodes/test_factory.py
View File

@ -11,7 +11,7 @@ from bonobo.util.testing import BufferingNodeExecutionContext
class FactoryTypeTest(TestCase): class FactoryTypeTest(TestCase):
def execute_node(self, node, *rows): def execute_node(self, node, *rows):
with BufferingNodeExecutionContext(node) as context: with BufferingNodeExecutionContext(node) as context:
context.write_sync(*map(Bag, rows)) context.write_sync(*rows)
return context.get_buffer() return context.get_buffer()
def test_args_as_str(self): def test_args_as_str(self):