scrivendo una funzione pytest per controllare l'output in un file in python?

Question

Ho chiesto alla domanda this su come scrivere un pytest per verificare l'output in stdout e ottenere una soluzione. Ora ho bisogno di scrivere un test case, per verificare se i contenuti sono scritte nel file e che i contenuti sono scritti come previsto esempio:

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

ora una funzione pytest per verificare se scritto:

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

mentre questo sembra funzionare, non penso che questo sia l'ideale, specialmente con hard-coding. in che modo questo tipo di test functions di scrittura su un file viene solitamente scritto?

Answer 1

C'è lo tmpdir fixture che creerà una directory temporanea per prova. Quindi un test sarebbe simile a questa:

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' 

Qui si sta refactoring del codice non da hard-coded o, che è un esempio lampante di come testare il codice fa a progettare meglio.