Add type hints

Author: suaaa7

python_test_example/flask_app/app_test.py

@@ -1,6 +1,6 @@
 import json
 from unittest import TestCase, main
-from unittest.mock import patch
+from unittest.mock import MagicMock, patch
 
 print('In app_test')
 
@@ -39,7 +39,7 @@ def test_predict_data(self):
         self.assertIsInstance(json.loads(response.data)['result'], float)
 
     @patch('flask_app.service.Service.check_model', return_value=False)
-    def test_predict_503(self, mock):
+    def test_predict_503(self, mock: MagicMock):
         response = self.client.post('/predict')
         self.assertEqual(response.status_code, 503)
 

python_test_example/flask_app/service.py

@@ -10,11 +10,11 @@ def __init__(self):
         self.model = load_model()
         print(f'self.model: {self.model}')
 
-    def predict(self, target: str):
+    def predict(self, target: str) -> float:
         print('Call predict in service')
         return randint(0, 1000) / 1000.0
 
-    def check_model(self):
+    def check_model(self) -> bool:
         if self.model:
             return True
         return False

python_test_example/mock/my_class.py

@@ -1,6 +1,6 @@
 import requests
 
 class MyClass:
-    def fetch_json(self, url: str):
+    def fetch_json(self, url: str) -> dict:
         response = requests.get(url)
         return response.json()

python_test_example/mock/my_class_test.py

@@ -1,4 +1,5 @@
-from unittest import TestCase, main, mock
+from unittest import TestCase, main
+from unittest.mock import MagicMock, call, patch
 
 from mock.my_class import MyClass
 
@@ -19,8 +20,8 @@ def json(self):
 
         return MockResponse(None, 404)
 
-    @mock.patch('requests.get', side_effect=mocked_requests_get)
-    def test_fetch_json(self, mock_get):
+    @patch('requests.get', side_effect=mocked_requests_get)
+    def test_fetch_json(self, mock_get: MagicMock):
         my_class = MyClass()
 
         json_data = my_class.fetch_json('http://example.com/test.json')
@@ -31,10 +32,10 @@ def test_fetch_json(self, mock_get):
         self.assertIsNone(json_data)
 
         self.assertIn(
-            mock.call('http://example.com/test.json'), mock_get.call_args_list
+            call('http://example.com/test.json'), mock_get.call_args_list
         )
         self.assertIn(
-            mock.call('http://example.com/another_test.json'), mock_get.call_args_list
+            call('http://example.com/another_test.json'), mock_get.call_args_list
         )
 
         self.assertEqual(len(mock_get.call_args_list), 3)