Try Mock using Decorator

try_mock_using_decorator.py

import random
import unittest
from unittest.mock import patch


class TryMockUsingDecoratorTest(unittest.TestCase):
    @patch('__main__.random.randrange')
    @patch('__main__.random.randint')
    def test_mock_random_twice(self, mock_randint, mock_randrange):
        try_mock_using_decorator()
        mock_randint.assert_called_once_with(1, 2)
        mock_randrange.assert_called_once_with(3, 4)


def try_mock_using_decorator():
    random.randint(1, 2)
    random.randrange(3, 4)


unittest.main()