Fix missing annotations.

ionelmc · Jan 5, 2016 · 495291a · 495291a
1 parent 6945b06
commit 495291a
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/src/pytest_benchmark/session.py b/src/pytest_benchmark/session.py
@@ -112,8 +112,7 @@ def prepare_benchmarks(self):
                         yield flat_bench
                 flat_bench = bench.as_dict(include_data=False, flat=True)
                 flat_bench["path"] = None
-                if compared:
-                    annotate_source(flat_bench, "NOW")
+                annotate_source(flat_bench, compared and "NOW")
                 yield flat_bench
 
     @property

diff --git a/src/pytest_benchmark/utils.py b/src/pytest_benchmark/utils.py
@@ -350,8 +350,11 @@ def annotate_source(bench, source):
     bench.update(
         canonical_name=bench["name"],
         canonical_fullname=bench["fullname"],
-        name="{0} ({1})".format(bench["name"], source),
-        fullname="{0} ({1})".format(bench["fullname"], source),
     )
-    bench["source"] = source
+    if source:
+        bench.update(
+            name="{0} ({1})".format(bench["name"], source),
+            fullname="{0} ({1})".format(bench["fullname"], source),
+        )
+        bench["source"] = source
     return bench