Updated tests

src/microbit/test/test_init.py

@@ -11,17 +11,35 @@
 
 class TestShim(object):
     def test_sleep(self):
+        # Save pointer to function about to be mocked
+        real_function = MicrobitModel.sleep
+
         milliseconds = 100
         MicrobitModel.sleep = mock.Mock()
         sleep(milliseconds)
         MicrobitModel.sleep.assert_called_with(milliseconds)
 
+        # Restore original function
+        MicrobitModel.sleep = real_function
+
     def test_running_time(self):
+        # Save pointer to function about to be mocked
+        real_function = MicrobitModel.running_time
+
         MicrobitModel.running_time = mock.Mock()
         running_time()
         MicrobitModel.running_time.assert_called_once()
 
+        # Restore original function
+        MicrobitModel.running_time = real_function
+
     def test_temperature(self):
+        # Save pointer to function about to be mocked
+        real_function = MicrobitModel.temperature
+
         MicrobitModel.temperature = mock.Mock()
         temperature()
         MicrobitModel.temperature.asser_called_once()
+
+        # Restore original function
+        MicrobitModel.temperature = real_function

src/microbit/test/test_microbit_model.py

@@ -21,7 +21,6 @@ def test_running_time(self):
         mock_end_time = 300
         self.__mb._MicrobitModel__start_time = mock_start_time
         time.time = mock.MagicMock(return_value=mock_end_time)
-        print(time.time())
         assert mock_end_time - mock_start_time == pytest.approx(
             self.__mb.running_time()
         )