在Unittest中测试生成文件和打印现有文件的函数

import unittest

def generate_file(file_path):
    with open(file_path, 'w') as file:
        file.write('This is a test file.')

def print_file(file_path):
    with open(file_path, 'r') as file:
        content = file.read()
        print(content)

class FileTestCase(unittest.TestCase):
    def setUp(self):
        self.file_path = 'test_file.txt'
        generate_file(self.file_path)

    def tearDown(self):
        # 删除测试文件
        os.remove(self.file_path)

    def test_generate_file(self):
        # 调用生成文件的函数
        generate_file('test_file.txt')
        
        # 验证文件是否存在
        self.assertTrue(os.path.exists('test_file.txt'))
        
        # 验证文件内容是否正确
        with open('test_file.txt', 'r') as file:
            content = file.read()
            self.assertEqual(content, 'This is a test file.')

    def test_print_file(self):
        # 调用打印文件的函数
        print_file(self.file_path)
        # 这里可以使用其他断言方法，如assertPrintedContentEqual()来验证打印输出的内容

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