Pretty print

Author: OliverRietmann

.github/scripts/csv_to_md.py

@@ -7,28 +7,36 @@
 parser.add_argument('-c', '--current', required=True, help='Current CSV file')
 args = parser.parse_args()
 
-pretty = lambda x : "{:.1f}".format(x) if x <= 0 else "+{:.1f}".format(x)
+pretty = lambda x : f'{x:.1f}'
 
-with open(args.baseline) as csv_file:
-  csv_reader = csv.reader(csv_file)
-  next(csv_reader)
-  table_baseline = [row for row in csv_reader]
+def get_emoji(x):
+  if x <= -1.0:
+    return ':green_circle:'
+  elif x >= 1.0:
+    return ':red_circle:'
+  else:
+    return ':white_circle:'
 
 with open(args.current) as csv_file:
   csv_reader = csv.reader(csv_file)
   next(csv_reader)
-  table_benchmark = [row for row in csv_reader]
+  table_current = [row for row in csv_reader]
+ 
+with open(args.baseline) as csv_file:
+  csv_reader = csv.reader(csv_file)
+  next(csv_reader)
+  table_baseline = [row for row in csv_reader]
 
 table = []
-for benchmark, baseline in zip(table_benchmark, table_baseline):
-  assert(benchmark[0] == baseline[0])
-  name = benchmark[0]
-  time = benchmark[1]
-  stdev = u"\u00B1" + str(benchmark[2])
-  d = float(baseline[1]) - float(benchmark[1])
-  emoji = ':red_circle:' if 0 < d else ':green_circle:'
-  difference = pretty(d)
+for current, baseline in zip(table_current, table_baseline):
+  assert(current[0] == baseline[0])
+  name = current[0]
+  time = int(float(current[1]))
+  stdev = pretty(float(current[2]))
+  d = float(baseline[1]) - float(current[1])
+  difference = int(d)
   percent = pretty(0 if float(baseline[1]) == 0.0 else 100 * d / float(baseline[1]))
+  emoji = get_emoji(float(percent))
   table.append([name, time, stdev, emoji, difference, percent])
 
 header = ["name", "time", "stdev", "", "difference", "percent"]