addCleanup vs tearDown

Question:

Recently, Ned Batchelder during his talk at PyCon 2016 noted:

If you are using unittest to write your tests, definitely use
addCleanup, it’s much better than tearDown.

Up until now, I’ve never used addCleanup() and got used to setUp()/tearDown() pair of methods for test “set up” and “tear down” phases.

Why should I switch to addCleanup() instead of tearDown()?


It was also recently discussed in the Python unittest with Robert Collins podcast.

Asked By: alecxe

||

Answers:

Per the addCleanup doc string:

Cleanup items are called even if setUp fails (unlike tearDown)

addCleanup can be used to register multiple functions, so you could use
separate functions for each resource you wish to clean up. That would allow your
code to be a bit more reusable/modular.

Answered By: unutbu

addCleanup() methods will run even if one of them fails, and will run even if setUp() fails. You should also consider using pytest.

Answered By: Raphaël Gomès

Another good thing about addCleanup is that it just works as you’d expect.

For example, if you call it in a setUp function, then all test methods will call the cleanup function in the end.

If you call it in a test method, only this method calls the cleanup function.

Answered By: laike9m