class TestDummyShutter(unittest.TestCase): def setUp(self): self.shutter = DummyShutter() def test_open(self): self.shutter.open().wait() self.assertEquals(self.shutter.state, self.shutter.OPEN) def test_close(self): self.shutter.close().wait() self.assertEquals(self.shutter.state, self.shutter.CLOSED)
class TestDummyShutter(TestCase): def setUp(self): super(TestDummyShutter, self).setUp() self.shutter = DummyShutter() def test_open(self): self.assertEqual(self.shutter.state, 'open') def test_close(self): self.shutter.close().join() self.assertEqual(self.shutter.state, 'closed')