viết một hàm pytest để kiểm tra xuất ra một tệp trong python?

Question

Tôi đã hỏi this câu hỏi về cách viết một pytest để kiểm tra kết xuất trong stdout và nhận giải pháp. Bây giờ tôi cần phải viết một test case, để kiểm tra xem các nội dung được ghi vào tập tin và nội dung được viết như mong đợi ví dụ:

def writetoafile(): 
    file = open("output.txt",w) 
    file.write("hello\n") 
    file.write("world\n") 
    file.close() 

bây giờ là một chức năng pytest để kiểm tra xem nó viết:

def test_writeToFile(): 
    file = open("ouput.txt",'r') 
    expected = "hello\nworld\n" 
    assert expected==file.read() 

trong khi điều này dường như hoạt động, tôi không nghĩ rằng đây là lý tưởng, đặc biệt là mã hóa cứng. làm thế nào là những loại test functions của văn bản cho một tập tin thường được viết?

Answer 1

Có tmpdir fixture sẽ tạo cho bạn một thư mục tạm thời cho mỗi kiểm tra. Vì vậy, một thử nghiệm sẽ trông giống như sau:

def writetoafile(fname): 
    with open(fname, 'w') as fp: 
     fp.write('Hello\n') 

def test_writetofile(tmpdir): 
    file = tmpdir.join('output.txt') 
    writetoafile(file.strpath) # or use str(file) 
    assert file.read() == 'Hello\n' 

Đây là ví dụ điển hình về cách kiểm tra mã giúp bạn thiết kế mã tốt hơn.