Merge pull request #74 from davegoopot/master

Fixing #73

lib/sleep.py

@@ -4,14 +4,18 @@ ___license___ = "MIT"
 
 import time
 
+
 def sleep_ms(duration):
-    # todo: deepsleep?
-    time.sleep_ms(duration)
+    start_time = time.ticks_ms()
+    end_time = start_time + duration
+    while time.ticks_ms() < end_time:
+        wfi()
+
 
 def sleep(duration):
-    # todo: deepsleep?
-    time.sleep(duration)
+    sleep_ms(duration * 1000)
+    
 
 def wfi():
     # todo: this is fake
-    sleep_ms(1)
+    time.sleep_ms(1)

lib/test_sleep.py

@@ -3,12 +3,24 @@
 ___license___      = "MIT"
 ___dependencies___ = ["upip:unittest", "sleep"]
 
-import unittest, sleep
+import unittest, sleep, time
 
 class TestSleep(unittest.TestCase):
 
     def test_sleep(self):
-        sleep.sleep_ms(100)
+        sleep_secs = 5
+        time_before = time.ticks_ms()
+        time_after = time_before + 1000 * sleep_secs
+        sleep.sleep(sleep_secs)
+        self.assertTrue(time.ticks_ms() >= time_after)
+
+    def test_sleep_ms(self):
+        sleep_ms = 3000
+        time_before = time.ticks_ms()
+        time_after = time_before + sleep_ms
+        sleep.sleep_ms(sleep_ms)
+        self.assertTrue(time.ticks_ms() >= time_after)
+
 
 if __name__ == '__main__':
     unittest.main()