Switched --env / -e optional argument from register to register_generic_run_arguements per suggestion. Also removed tests/util/get_passed_env.py

bonobo/commands/run.py

@@ -106,6 +106,7 @@ def register_generic_run_arguments(parser, required=True):
     source_group = parser.add_mutually_exclusive_group(required=required)
     source_group.add_argument('filename', nargs='?', type=str)
     source_group.add_argument('--module', '-m', type=str)
+    parser.add_argument('--env', '-e', action='append')
     return parser
 
 
@@ -115,5 +116,4 @@ def register(parser):
     verbosity_group.add_argument('--quiet', '-q', action='store_true')
     verbosity_group.add_argument('--verbose', '-v', action='store_true')
     parser.add_argument('--install', '-I', action='store_true')
-    parser.add_argument('--env', '-e', action='append')
     return execute

tests/util/get_passed_env.py

@@ -1,22 +0,0 @@
-import os
-
-from bonobo import Graph
-
-
-def extract():
-    env_test_user = os.getenv('ENV_TEST_USER')
-    env_test_number = os.getenv('ENV_TEST_NUMBER')
-    env_test_string = os.getenv('ENV_TEST_STRING')
-    return env_test_user, env_test_number, env_test_string
-
-
-def load(s: str):
-    print(s)
-
-
-graph = Graph(extract, load)
-
-if __name__ == '__main__':
-    from bonobo import run
-
-    run(graph)