A4

Author: whb07

__init__.py

@@ -0,0 +1,11 @@
+from time import sleep
+
+
+def get_water_level_africa():
+    sleep(15)
+    return '15 cm'
+
+
+def get_future_rainfall():
+    rain_level = get_water_level_africa().split(' ')[0]
+    return rain_level

africa_water.py

@@ -0,0 +1,13 @@
+import pytest
+from unittest import mock
+from ..africa_water import get_future_rainfall
+
+# def test_water_response():
+#     response = get_water_level_africa()
+#     assert 'cm' in response, 'should be in cm'
+
+
+@mock.patch('testing_python_meetup.africa_water.get_water_level_africa')
+def test_mock_water_response(mock_response):
+    mock_response.return_value = '25 cm'
+    assert '25' in mock_response(), 'should be in cm'