Fix unit test for genElapsedTime.py for header change

Signed-off-by: Matt Liberty <[email protected]>

Author: maliberty

flow/test/test_genElapsedTime.py

@@ -26,7 +26,8 @@ def test_elapsed_time(self, mock_stdout):
             f.write("Some log entry\n")
             f.write("Elapsed time: 01:30:00[h:]min:sec.\n")
         # call the script with the test log file
-        sys.argv = ["./genElapsedTime.py", "--logDir", str(self.tmp_dir.name)]
+        sys.argv = ["./genElapsedTime.py", "--logDir", str(self.tmp_dir.name),
+                    "--noHeader"]
         with patch.object(sys, 'argv', sys.argv):
             module = importlib.import_module(self.module_name)
         # check if output is correct
@@ -40,7 +41,8 @@ def test_zero_time(self, mock_stdout):
             f.write("Some log entry\n")
             f.write("Elapsed time: 00:00:74[h:]min:sec.\n")
         # call the script with the test log file
-        sys.argv = ["./genElapsedTime.py", "--logDir", str(self.tmp_dir.name)]
+        sys.argv = ["./genElapsedTime.py", "--logDir", str(self.tmp_dir.name),
+                    "--noHeader"]
         with patch.object(sys, 'argv', sys.argv):
             module = importlib.import_module(self.module_name)
         # check if output is correct
@@ -53,7 +55,8 @@ def test_elapsed_time_longer_duration(self, mock_stdout):
             f.write("Some log entry\n")
             f.write("Elapsed time: 12:24.14[h:]min:sec. CPU time: user 5081.82 sys 170.18 (705%). Peak memory: 9667132KB.\n")
         # call the script with the test log file
-        sys.argv = ["util/genElapsedTime.py", "--logDir", str(self.tmp_dir.name)]
+        sys.argv = ["util/genElapsedTime.py", "--logDir", str(self.tmp_dir.name),
+                    "--noHeader"]
         with patch.object(sys, 'argv', sys.argv):
             module = importlib.import_module(self.module_name)
             importlib.reload(module)
@@ -72,7 +75,9 @@ def test_missing_arg(self):
     def test_no_elapsed_time(self):
         with open(self.log_file, "w") as f:
             f.write('Other log message')
-        with patch('sys.argv', ["util/genElapsedTime.py", "--logDir", str(self.tmp_dir.name)]):
+        with patch('sys.argv', ["util/genElapsedTime.py",
+                                "--logDir", str(self.tmp_dir.name),
+                                "--noHeader"]):
             with patch('sys.stderr', new=StringIO()) as fake_err_output:
                 module = importlib.import_module(self.module_name)
                 importlib.reload(module)

flow/util/genElapsedTime.py

@@ -15,6 +15,8 @@
     description='Print elapsed time for every step in the flow')
 parser.add_argument('--logDir', '-d', required=True,
                     help='Log files directory')
+parser.add_argument('--noHeader', action='store_true',
+                    help='Skip the header')
 args = parser.parse_args()
 
 if not args.logDir:
@@ -57,7 +59,7 @@
 
     # Print the name of the step and the corresponding elapsed time
     if elapsedTime != 0:
-        if first:
+        if first and not args.noHeader:
             print("%-25s %10s" % ("Log", "Elapsed seconds"))
             first = False
         print('%-25s %10s' % (os.path.splitext(os.path.basename(str(f)))[0], elapsedTime))