unittest.mock Final QA (#550)

* Final QA

* Final QA on README

* Sort imports according to Ruff

Author: gahjelle

unittest-mock/README.md

@@ -2,7 +2,7 @@
 
 This folder contains code examples related to the RealPython tutorial on [Understanding the Python Mock Object Library](https://realpython.com/python-mock-library/).
 
-The example code showcases some of the different use cases of `unittest.mock` in a single test file, `test_my_calendar.py`. When writing your own tests, keep in mind that readability counts and that your test code will be more readable if you keep one consistent approach to mocking.
+The example code showcases some of the different use cases of `unittest.mock` in a single test file, `test_holidays.py`. When writing your own tests, keep in mind that readability counts and that your test code will be more readable if you keep one consistent approach to mocking.
 
 ## Installation
 
@@ -23,7 +23,7 @@ $ source venv/bin/activate
 ## Run the Tests
 
 ```sh
-(venv) $ python test_my_calendar.py
+(venv) $ python test_holidays.py
 ```
 
 All the tests should pass. Go ahead and play with the code examples, change them, break them, and practice your understanding of using `unittest.mock` for testing in Python.
@@ -34,4 +34,4 @@ Martin Breuss - Email: [email protected]
 
 ## License
 
-Distributed under the MIT license. See ``LICENSE`` for more information.
+Distributed under the MIT license. See `LICENSE` for more information.

unittest-mock/my_calendar.py unittest-mock/holidays.pyunittest-mock/my_calendar.py renamed to unittest-mock/holidays.py

@@ -1,38 +1,37 @@
-import datetime
 import unittest
+from datetime import datetime
 from unittest.mock import Mock, patch
 
 import requests
-from my_calendar import get_holidays, is_weekday
+from holidays import get_holidays, is_weekday
 from requests.exceptions import Timeout
 
 
 class TestCalendar(unittest.TestCase):
     @classmethod
     def setUpClass(cls):
-        cls.wednesday = datetime.datetime(year=2025, month=1, day=1)
-        cls.sunday = datetime.datetime(year=2025, month=1, day=5)
+        cls.wednesday = datetime(year=2025, month=1, day=1)
+        cls.sunday = datetime(year=2025, month=1, day=5)
         cls.holidays = {"12/25": "Christmas", "7/4": "Independence Day"}
         cls.response_setup_dict = {
-            "json.return_value": TestCalendar.holidays,
+            "json.return_value": cls.holidays,
             "status_code": 200,
         }
 
     def log_request(self, url):
         """Helper function that logs and returns a mock successful request."""
         print(f"Making a request to {url}.")
         print("Request received!")
-        response_mock = Mock(**TestCalendar.response_setup_dict)
-        return response_mock
+        return Mock(**self.response_setup_dict)
 
-    @patch("my_calendar.datetime")
+    @patch("holidays.datetime")
     def test_is_weekday_returns_true_on_weekdays(self, mock_datetime):
-        mock_datetime.today.return_value = TestCalendar.wednesday
+        mock_datetime.today.return_value = self.wednesday
         self.assertTrue(is_weekday())
 
-    @patch("my_calendar.datetime")
+    @patch("holidays.datetime")
     def test_is_weekday_returns_false_on_weekends(self, mock_datetime):
-        mock_datetime.today.return_value = TestCalendar.sunday
+        mock_datetime.today.return_value = self.sunday
         self.assertFalse(is_weekday())
 
     # Example of patching only a specific object
@@ -43,13 +42,13 @@ def test_get_holidays_timeout(self, mock_requests):
 
     # Example of using the `with` statement with `patch()`
     def test_get_holidays_logging(self):
-        with patch("my_calendar.requests") as mock_requests:
+        with patch("holidays.requests") as mock_requests:
             mock_requests.get.side_effect = self.log_request
             self.assertEqual(get_holidays()["12/25"], "Christmas")
 
-    @patch("my_calendar.requests")
+    @patch("holidays.requests")
     def test_get_holidays_retry(self, mock_requests):
-        response_mock = Mock(**TestCalendar.response_setup_dict)
+        response_mock = Mock(**self.response_setup_dict)
         # Set the side effect of .get()
         mock_requests.get.side_effect = [Timeout, response_mock]
         # Test that the first request raises a Timeout