Stop running tests if setUp raises an exception in Python unittest

Question

I have this test class:

class mytest(unittest.TestCase):
    def setUp(self):
        os.mkdir(...)
        ...

    def tearDown(self):
        shutil.rmtree(...)

    def test_one(self):
        ...

    def test_two(self):
        ...

If something fails after mkdir has ran when running setUp of test_one, it will still try to run setUp of test_two. At this point I'll get an error on mkdir because rmtree didn't run.

Is there any way to tell Python unittest to stop running the current test if setUp fails? I'm not looking to stop on a regular test failure.

Answer 1

Add a failure call in the setUp method.

def setUp(self):
    try:
        somethingThatMightFail()
    except:
        self.fail()