napisanie funkcji pytest, aby sprawdzić wyjście do pliku w python?

Question

Zapytałem this pytanie o to, jak napisać monit, aby sprawdzić wyjście w stdout i otrzymałem rozwiązanie. Teraz muszę napisać test case, aby sprawdzić, czy zawartość są zapisywane w pliku i że zawartość są zapisywane zgodnie z oczekiwaniami np

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

teraz funkcją pytest aby sprawdzić, czy to napisane:

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

podczas gdy to wydaje się działać, nie sądzę, że jest to idealne, zwłaszcza trudne kodowanie. w jaki sposób zapisuje się ten typ pliku w typowym pliku test functions?

Answer 1

Istnieje tmpdir fixture, który utworzy tymczasowy katalog testowy. Więc test będzie wyglądać mniej więcej tak:

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' 

Tutaj jesteś refaktoryzacji kodu nie może być ustalony albo, co jest najlepszym przykładem tego, jak testowanie kodu sprawia, że ​​można zaprojektować go lepiej.