[cli] Fix "AttributeError" of pytest arguments (#1505)

taichi-dev · Jul 16, 2020 · 4db65b0 · 4db65b0
1 parent 43c32b8
commit 4db65b0
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/python/taichi/main.py b/python/taichi/main.py
@@ -715,19 +715,19 @@ def _test_python(args):
             pytest_args += ['-s', '-v']
         if args.rerun:
             pytest_args += ['--reruns', args.rerun]
-        if args.keys:
-            pytest_args += ['-k', args.keys]
-        if args.marks:
-            pytest_args += ['-m', args.marks]
-        if args.failed_first:
-            pytest_args += ['--failed-first']
-        if args.fail_fast:
-            pytest_args += ['--exitfirst']
         try:
             if args.coverage:
                 pytest_args += ['--cov-branch', '--cov=python/taichi']
             if args.cov_append:
                 pytest_args += ['--cov-append']
+            if args.keys:
+                pytest_args += ['-k', args.keys]
+            if args.marks:
+                pytest_args += ['-m', args.marks]
+            if args.failed_first:
+                pytest_args += ['--failed-first']
+            if args.fail_fast:
+                pytest_args += ['--exitfirst']
         except AttributeError:
             pass