unittest_sample4.py

import unittest
from unittest.mock import patch
from sample_method import SampleMethod


class TestSampleMethod(unittest.TestCase):
    def setUp(self):
        print('setUp')
        self.sample = SampleMethod()

    def tearDown(self):
        print('tearDown')

    def test_method_1(self):
        print('test_method_1')
        with patch('sample_method.SampleMethod.method_1') as mock_method_1:
            mock_method_1.return_value = 'AAA'
            actual = self.sample.method_1()
            print(self.sample.method_1())  # AAA
        print(self.sample.method_1())  # method_1
        expected = 'AAA'
        self.assertEqual(expected, actual)


if __name__ == '__main__':
    # unittestを実行
    unittest.main()