FIX argument propogation bug

modpipe/__init__.py

@@ -3,4 +3,4 @@
 
 __author__ = """John Bjorn Nelson"""
 __email__ = 'jbn@abreka.com'
-__version__ = '0.0.2'
+__version__ = '0.0.3'

modpipe/modpipe_impl.py

@@ -217,7 +217,12 @@ def apply_to(self, f, arity):
         else:
             res = f(self.args)
 
-        return res if isinstance(res, Result) else Result(res)
+        if isinstance(res, Result):
+            return res
+        elif res is not None:
+            return Result(res)
+        else:
+            return Result(self.args)
 
 
 class Done(Result):

setup.py

@@ -37,6 +37,6 @@
     test_suite='tests',
     tests_require=test_requirements,
     url='https://github.com/jbn/modpipe',
-    version='0.0.2',
+    version='0.0.3',
     zip_safe=False,
 )

tests/examples/inconsistent_pipeline.py

@@ -0,0 +1,14 @@
+def f(x):
+    return {'value': x}
+
+
+def g(src):
+    return src, {}
+
+
+def h(src, dst):
+    dst['computation'] = src['value'] * 2
+
+
+def last_one(src, dst):
+    return dst

tests/test_regressions.py

@@ -0,0 +1,8 @@
+from modpipe import ModPipe
+
+
+def test_inconsistent_pipeline():
+    pipe = ModPipe('tests.examples.inconsistent_pipeline')
+    res = pipe(1)
+
+    assert res == {'computation': 2}