From 8dda00c46e73f2a44e7c387a6b4e86055ffecea2 Mon Sep 17 00:00:00 2001
From: J08nY
Date: Mon, 18 Mar 2019 19:37:51 +0100
Subject: Add tests to Jupyter notebooks.

---
 util/utils.py | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

(limited to 'util/utils.py')

diff --git a/util/utils.py b/util/utils.py
index d6b9aed..5359988 100644
--- a/util/utils.py
+++ b/util/utils.py
@@ -16,6 +16,21 @@ def moving_average(a, n) :
     return ret[n - 1:] / n
 
 
+def time_scale(data, orig_unit, target_unit, scaling_factor):
+    units = {
+        "milli": ("ms", 1000000),
+        "micro": (r"$\mu s$", 1000),
+        "nano":  ("ns", 1)
+    }
+    upper = units[orig_unit][1]
+    lower = units[target_unit][1] * scaling_factor
+    if upper > lower:
+        data *= upper // lower
+    elif lower > upper:
+        np.floor_divide(data, lower // upper, data)
+    return (r"$\frac{1}{" + str(scaling_factor) + "}$" if scaling_factor != 1 else "") + units[target_unit][0]
+
+
 def plot_hist(axes, data, xlabel=None, log=False, avg=True, median=True, bins=None, **kwargs):
     time_max = max(data)
     time_min = min(data)
-- 
cgit v1.2.3-70-g09d2