class TestPlane(unittest.TestCase):
def setUp(self):
self.plane = Plane()
def test_isflying_should_be_true_on_initialization(self):
self.assertTrue(self.plane.isflying)
def test_land_should_change_isflying_to_false(self):
self.plane.land()
self.assertFalse(self.plane.isflying)
def test_take_off_should_change_isflying_to_true(self):
self.plane.land()
self.plane.take_off()
self.assertTrue(self.plane.isflying)
def test_take_off_should_fail_if_plane_not_landed(self):
with self.assertRaisesRegexp(Exception, 'Plane cannot take off while flying'):
self.plane.take_off()
class TestPlane(unittest.TestCase):
def setUp(self):
self.plane = Plane()
def test_isflying_should_be_true_on_initialization(self):
self.assertTrue(self.plane.isflying)
def test_land_should_change_isflying_to_false(self):
self.plane.land()
self.assertFalse(self.plane.isflying)
def test_take_off_should_change_isflying_to_true(self):
self.plane.land()
self.plane.take_off()
self.assertTrue(self.plane.isflying)
def test_take_off_should_fail_if_plane_not_landed(self):
with self.assertRaisesRegexp(Exception,
'Plane cannot take off while flying'):
self.plane.take_off()
def setUp(self):
self.airport = Airport()
self.plane = Plane()
def setUp(self):
self.plane = Plane()
def setUp(self):
self.plane = Plane()
def load(self):
for plane_model in session.query(PlaneModel).all():
self.logger.debug(f'Loading plane: {plane_model} ...')
self.planes[plane_model.id] = Plane(plane_model, self)
self.add_task(self.logger.debug, 100, '100 ticks have passed')
