For Python unit testing, using a context manager instead of setUp() and tearDown()

test.py

# For Python unit testing, an idea for how to use a context manager for
# a unittest.TestCase instead of the setUp() and tearDown() methods.
#
# This is a possible answer to--
#
#   http://stackoverflow.com/questions/8416208/in-python-is-there-a-good-idiom-for-using-context-managers-in-setup-teardown

from contextlib import contextmanager
import unittest

@contextmanager
def resource_manager():
    yield 'foo'

class MyTest(unittest.TestCase):

    def run(self, result=None):
        with resource_manager() as resource:
            self.resource = resource
            super(MyTest, self).run(result)

    def test(self):
        self.assertEqual('foo', self.resource)

unittest.main()

Thanks! :)