rely on utime directly.

[hackapet/Adafruit_Blinka.git] / python / testing / profiling / __init__.py

diff --git a/python/testing/profiling/__init__.py b/python/testing/profiling/__init__.py
index 0d584f64b1e0d2faa6e36bcef5af3974ff07e7d3..fce14a41c14fbdd4a038c2e2b4fdc81d706c1a47 100644 (file)
--- a/python/testing/profiling/__init__.py
+++ b/python/testing/profiling/__init__.py
@@ -1,11 +1,11 @@
 def time_function(f):
-    from agnostic import time
+    import utime
     """From http://docs.micropython.org/en/latest/esp8266/reference/speed_python.html"""
     myname = str(f).split(' ')[1]
     def new_func(*args, **kwargs):
-        t = time.ticks_us()
+        t = utime.ticks_us()
         result = f(*args, **kwargs)
-        delta = time.ticks_diff(time.ticks_us(), t)
+        delta = utime.ticks_diff(utime.ticks_us(), t)
         print('Function {} Time = {:6.3f}ms'.format(myname, delta/1000))
         return result
     return new_func