def test_long_case(self):
ticket_1 = Ticket("PIT", "ORD")
ticket_2 = Ticket("XNA", "SAP")
ticket_3 = Ticket("SFO", "BHM")
ticket_4 = Ticket("FLG", "XNA")
ticket_5 = Ticket("NONE", "LAX")
ticket_6 = Ticket("LAX", "SFO")
ticket_7 = Ticket("SAP", "SLC")
ticket_8 = Ticket("ORD", "NONE")
ticket_9 = Ticket("SLC", "PIT")
ticket_10 = Ticket("BHM", "FLG")
tickets = [
ticket_1, ticket_2, ticket_3, ticket_4, ticket_5, ticket_6,
ticket_7, ticket_8, ticket_9, ticket_10
]
expected = [
"LAX", "SFO", "BHM", "FLG", "XNA", "SAP", "SLC", "PIT", "ORD",
"NONE"
]
result = reconstruct_trip(tickets, 10)
print(result)
self.assertTrue(expected == result)
def test_short_case(self):
ticket_1 = Ticket("NONE", "PDX")
ticket_2 = Ticket("PDX", "DCA")
ticket_3 = Ticket("DCA", "NONE")
tickets = [ticket_1, ticket_2, ticket_3]
expected = ["PDX", "DCA", "NONE"]
result = reconstruct_trip(tickets, 3)
self.assertTrue(expected == result)
def test_incorrect_case(self):
incorrect_set = [
('LHD', 'DAB'),
(None, 'HVN'),
('MSO', 'SFO'),
('RDU', 'ABQ'),
('ACY', None),
]
self.assertEqual(reconstruct_trip(incorrect_set), [])
def test_long_case(self):
long_set = [
('PIT', 'ORD'),
('XNA', 'CID'),
('SFO', 'BHM'),
('FLG', 'XNA'),
(None, 'LAX'),
('LAX', 'SFO'),
('CID', 'SLC'),
('ORD', None),
('SLC', 'PIT'),
('BHM', 'FLG'),
]
self.assertEqual(
reconstruct_trip(long_set),
['LAX', 'SFO', 'BHM', 'FLG', 'XNA', 'CID', 'SLC', 'PIT', 'ORD'])
def test_short_case(self):
short_set = [(None, 'PDX'), ('PDX', 'DCA'), ('DCA', None)]
self.assertEqual(reconstruct_trip(short_set), ['PDX', 'DCA'])
