wip: fix aiidateam#700 but wait for pre-commit being merged

DropD · Sep 20, 2017 · 55772c1 · 55772c1
1 parent 75fb4a8
commit 55772c1
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/aiida/cmdline/tests/test_verdi_help.py b/aiida/cmdline/tests/test_verdi_help.py
@@ -15,6 +15,8 @@ def test_verdi_help_full_string(self):
         with captured_output() as (out, err):
             try:
                 self.help_cmd.run()
+            except SystemExit:
+                pass
             finally:
-                output = out.getvalue().strip()
-                self.assertIn(output, '* import        Import nodes and group of nodes')
+                output = [l.strip() for l in out.getvalue().split('\n')]
+                self.assertIn('* import        Import nodes and group of nodes', output)
diff --git a/aiida/cmdline/verdilib.py b/aiida/cmdline/verdilib.py
@@ -1204,7 +1204,7 @@ def exec_from_cmdline(argv):
             v += "\n"
             # resilient_parsing suppresses interactive prompts
             v += cmd._ctx(args=[], resilient_parsing=True).get_help()
-            v = v.split('\n')  # need list of lines
+        v = v.split('\n')  # need list of lines
 
         lines = [l.strip() for l in v]
         empty_lines = [bool(l) for l in lines]