5.2. Test Patch

5.2.1. SetUp

>>> from unittest import TestCase
>>> from unittest.mock import patch

5.2.2. Patch Randint

Code:

>>> from random import randint
>>>
>>> def throw_dice():
...     return randint(1,6)

Test:

>>>
>>> with patch('random.randint', return_value=4) as randint:
...     assert throw_dice() == 4

5.2.3. Path Date

>>> def test_birthdate_age(self):
...     mark = User('Mark', 'Watney', birthdate='2000-01-02')
...     with patch('main.date') as d:
...         d.today.return_value = date(2024, 1, 2)
...         age = mark.get_age()
...     self.assertEqual(age, 24)