def test_create_horizontal(self): pos_1 = Position(5, 5) pos_2 = Position(7, 7) tunnel = Tunnel.create_horizontal(pos_1, pos_2) self.assertEquals(tunnel.pos.point, pos_1.point) self.assertEquals(tunnel.length, 2) self.assertEquals(tunnel.orient, Orientation.right)