import unittest

class Stage:
    def __init__(self):
        self.a = 5
        pass

s = Stage()

class UnitTestTest(unittest.TestCase):
    a = 5
    def setUp(self):
        if (s.a == 5):
            s.a = 0
    def tearDown(self):
        s.a = 1
        print "\ntearing down\n"
    def testA(self):
        self.assertEqual(s.a, 0)
    def testB(self):
        self.assertEqual(s.a, 1)

if __name__ == "__main__":
      unittest.main() # run all tests