如何在python测试中模拟subprocess.check_call异常

import unittest
from unittest import mock
import subprocess

def my_function():
    subprocess.check_call(['command', 'arg1', 'arg2'])

class MyTestCase(unittest.TestCase):
    @mock.patch('subprocess.check_call')
    def test_my_function_raises_exception(self, mock_check_call):
        mock_check_call.side_effect = subprocess.CalledProcessError(returncode=1, cmd='command')
        
        with self.assertRaises(subprocess.CalledProcessError):
            my_function()

if __name__ == '__main__':
    unittest.main()