Improved functionality of the -f option

testflo/filters.py

@@ -1,5 +1,8 @@
 from __future__ import print_function
 
+import os
+
+
 class TimeFilter(object):
     """This iterator saves to the specified output file only those tests
     that complete successfully in max_time seconds or less.  This output
@@ -29,13 +32,17 @@ def __init__(self, outfile):
         self.outfile = outfile
 
     def get_iter(self, input_iter):
-        fails = []
+        try:
+            os.remove(self.outfile)
+        except OSError:
+            pass
+
         for result in input_iter:
             if result.status == 'FAIL' and not result.expected_fail:
-                fails.append(result.spec)
-            yield result
-
-        if fails:
-            with open(self.outfile, 'w') as f:
-                for spec in sorted(fails):
+                with open(self.outfile, 'a') as f:
+                    if result.nprocs > 1:
+                        spec = f"{result.spec}  # mpi, nprocs={result.nprocs}"
+                    else:
+                        spec = result.spec
                     print(spec, file=f)
+            yield result

testflo/test.py

@@ -51,6 +51,8 @@ def testcontext(test):
     global _testing_path
     old_sys_path = sys.path
 
+    os.environ['TESTFLO_SPEC'] = test.spec
+
     _testing_path[0] = test.test_dir
     sys.path = _testing_path
 
@@ -61,6 +63,8 @@ def testcontext(test):
         test.err_msg = traceback.format_exc()
     finally:
         sys.path = old_sys_path
+        if 'TESTFLO_SPEC' in os.environ:
+            del os.environ['TESTFLO_SPEC']
 
 
 class Test(object):