aboutsummaryrefslogtreecommitdiffstats
path: root/tests/test_partitioner.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/test_partitioner.py')
-rwxr-xr-xtests/test_partitioner.py7
1 files changed, 4 insertions, 3 deletions
diff --git a/tests/test_partitioner.py b/tests/test_partitioner.py
index 5634c01..439f84c 100755
--- a/tests/test_partitioner.py
+++ b/tests/test_partitioner.py
@@ -20,7 +20,8 @@ sys.path.insert(0, glob.glob('./lib')[0])
from mapper import *
from partitioner import *
-def main(argv):
+def main():
+ argv = sys.argv[1:]
try:
#tosca = ToscaTemplate(argv[0])
tpl = yaml.load(file(argv[0],'r'))
@@ -52,6 +53,6 @@ def main(argv):
except:
print('Unexpected error: %s', sys.exc_info()[0])
raise
-if __name__ == "__main__":
- main(sys.argv[1:])
+if __name__ == "__main__":
+ sys.exit(main())