def test_reservation_number(self): ''' Test for reservation number ''' ticket = AmtrakTicket("ticket.pdf") self.assertEqual(ticket.reservation_number(), "366B20")
def test_origin(self): ''' Test for origin statation abbreviation ''' ticket = AmtrakTicket("ticket.pdf") self.assertEqual(ticket.origin(), "LAX")