how to delete a test file after finished testing in python?

Question

I create a test where in setUp I create file like so :

 class TestSomething :
     def setUp(self):
         # create file
         fo = open('some_file_to_test','w')
         fo.write('write_something')
         fo.close()

     def test_something(self):
         # call some function to manipulate file
         ...
         # do some assert
         ...

     def test_another_test(self):
         # another testing with the same setUp file
         ...

at the end of testing, regardless either succeed or not, I want the testing file gone,so how to delete the file after i have finished testing?

Answer 1

Assuming you're using a unittest-esque framework (i.e. nose, etc.), you would want to use the tearDown method to delete the file, as that will run after each test.

def tearDown(self):
    os.remove('some_file_to_test')

If you want to only delete this file after all the tests, you could create it in the method setUpClass and delete it in the method tearDownClass, which will run before and after all tests have run, respectively.

Answer 2

Write a tearDown method:

https://docs.python.org/3/library/unittest.html#unittest.TestCase.tearDown

def tearDown(self):
    import os
    os.remove('some_file_to_test')

Also have a look at the tempfile module and see if it's useful in this case.