print durations

python · giampaolo · Apr 2, 2023 · Mar 10, 2019 · Mar 10, 2019 · Mar 10, 2019
commit f8a49d3bd4075fe128ae564adc0396da56f6812c
diff --git a/Lib/unittest/case.py b/Lib/unittest/case.py
@@ -675,6 +675,8 @@ def run(self, result=None):
             return result
         finally:
             result.stopTest(self)
+            if result.durations is not None:
+                result.addDuration(self, result.runTime)
             if orig_result is None:
                 stopTestRun = getattr(result, 'stopTestRun', None)
                 if stopTestRun is not None:

diff --git a/Lib/unittest/result.py b/Lib/unittest/result.py
@@ -43,6 +43,7 @@ def __init__(self, stream=None, descriptions=None, verbosity=None):
         self.skipped = []
         self.expectedFailures = []
         self.unexpectedSuccesses = []
+        self.collectedDurations = []
         self.shouldStop = False
         self.buffer = False
         self.tb_locals = False
@@ -157,6 +158,9 @@ def addUnexpectedSuccess(self, test):
         """Called when a test was expected to fail, but succeed."""
         self.unexpectedSuccesses.append(test)
 
+    def addDuration(self, test, elapsed):
+        self.collectedDurations.append((test, elapsed))
+
     def wasSuccessful(self):
         """Tells whether or not this result was a success."""
         # The hasattr check is for test_result's OldResult test.  That

diff --git a/Lib/unittest/runner.py b/Lib/unittest/runner.py
@@ -45,7 +45,7 @@ def __init__(self, stream, descriptions, verbosity, *, durations=None):
 
     def _fmtRes(self, s):
         if self.durations is not None:
-            return "%.2f %s" % (self.runTime, s)
+            return "%.3f %s" % (self.runTime, s)
         else:
             return s
 
@@ -163,6 +163,18 @@ def _makeResult(self):
             return self.resultclass(self.stream, self.descriptions,
                                     self.verbosity)
 
+    def _printDurations(self, result):
+        self.stream.writeln("Slowest test durations")
+        if hasattr(result, 'separator2'):
+            self.stream.writeln(result.separator2)
+        ls = result.collectedDurations
+        ls.sort(key=lambda x: x[1])
+        if self.durations > 0:
+            ls = ls[-self.durations:]
+        for test, elapsed in ls:
+            print("%-12s %s" % ("%.3fs" % elapsed, str(test)))
+        self.stream.writeln("")
+
     def run(self, test):
         "Run the given test case or test suite."
         result = self._makeResult()
@@ -196,6 +208,8 @@ def run(self, test):
             stopTime = time.perf_counter()
         timeTaken = stopTime - startTime
         result.printErrors()
+        if getattr(self, 'durations', None) is not None:
+            self._printDurations(result)
         if hasattr(result, 'separator2'):
             self.stream.writeln(result.separator2)
         run = result.testsRun