How to a run a specific code before & after each unit test in python

Following is the structure of my tests in a file.
Class
 setup
  test01
  test02
  test03
 teardown

I have a requirement to run a specific code before and after each test.

For before, I could invoke that code from the setup. But for after the test, I am not able to figure how to do it. Obviously invoking the code from teardown would work for the last test, but how to have it run for the tests in between.