def _check_train_delay_disruption(dis):
is_valid_disruption(dis, chaos_disrup=False)
assert dis['disruption_id'] == 'vjA_delayed'
assert dis['severity']['effect'] == 'SIGNIFICANT_DELAYS'
assert len(dis['impacted_objects']) == 1
ptobj = dis['impacted_objects'][0]['pt_object']
assert ptobj['embedded_type'] == 'trip'
assert ptobj['id'] == 'vjA'
assert ptobj['name'] == 'vjA'
# for delay we should have detail on the impacted stops
impacted_objs = get_not_null(dis['impacted_objects'][0], 'impacted_stops')
assert len(impacted_objs) == 2
imp_obj1 = impacted_objs[0]
assert get_valid_time(get_not_null(imp_obj1, 'amended_arrival_time')) == _dt(h=8, m=2, s=24)
assert get_valid_time(get_not_null(imp_obj1, 'amended_departure_time')) == _dt(h=8, m=2, s=25)
assert get_not_null(imp_obj1, 'cause') == 'cow on tracks'
assert get_not_null(imp_obj1, 'departure_status') == 'delayed'
assert get_not_null(imp_obj1, 'arrival_status') == 'delayed'
assert get_not_null(imp_obj1, 'stop_time_effect') == 'delayed'
assert get_valid_time(get_not_null(imp_obj1, 'base_arrival_time')) == _dt(8, 1, 0)
assert get_valid_time(get_not_null(imp_obj1, 'base_departure_time')) == _dt(8, 1, 0)
imp_obj2 = impacted_objs[1]
assert get_valid_time(get_not_null(imp_obj2, 'amended_arrival_time')) == _dt(h=8, m=4, s=0)
assert get_valid_time(get_not_null(imp_obj2, 'amended_departure_time')) == _dt(h=8, m=4, s=0)
assert imp_obj2['cause'] == ''
assert get_not_null(imp_obj1, 'stop_time_effect') == 'delayed'
assert get_not_null(imp_obj1, 'departure_status') == 'delayed'
assert get_not_null(imp_obj1, 'arrival_status') == 'delayed'
assert get_valid_time(get_not_null(imp_obj2, 'base_departure_time')) == _dt(8, 1, 2)
assert get_valid_time(get_not_null(imp_obj2, 'base_arrival_time')) == _dt(8, 1, 2)
def _check_train_delay_disruption(dis):
is_valid_disruption(dis, chaos_disrup=False)
eq_(dis['disruption_id'], 'vjA_delayed')
eq_(dis['severity']['effect'], 'SIGNIFICANT_DELAYS')
eq_(len(dis['impacted_objects']), 1)
ptobj = dis['impacted_objects'][0]['pt_object']
eq_(ptobj['embedded_type'], 'trip')
eq_(ptobj['id'], 'vjA')
eq_(ptobj['name'], 'vjA')
# for delay we should have detail on the impacted stops
impacted_objs = get_not_null(dis['impacted_objects'][0], 'impacted_stops')
assert len(impacted_objs) == 2
imp_obj1 = impacted_objs[0]
eq_(get_valid_time(get_not_null(imp_obj1, 'amended_arrival_time')), _dt(h=8, m=2, s=24))
eq_(get_valid_time(get_not_null(imp_obj1, 'amended_departure_time')), _dt(h=8, m=2, s=25))
eq_(get_not_null(imp_obj1, 'cause'), 'cow on tracks')
# for the moment we output 00000, but we should output the right base departure/arrival
eq_(get_valid_time(get_not_null(imp_obj1, 'base_arrival_time')), _dt(8, 1, 0))
eq_(get_valid_time(get_not_null(imp_obj1, 'base_departure_time')), _dt(8, 1, 0))
imp_obj2 = impacted_objs[1]
eq_(get_valid_time(get_not_null(imp_obj2, 'amended_arrival_time')), _dt(h=8, m=4, s=0))
eq_(get_valid_time(get_not_null(imp_obj2, 'amended_departure_time')), _dt(h=8, m=4, s=0))
eq_(imp_obj2['cause'], '')
eq_(get_valid_time(get_not_null(imp_obj2, 'base_departure_time')), _dt(8, 1, 2))
eq_(get_valid_time(get_not_null(imp_obj2, 'base_arrival_time')), _dt(8, 1, 2))
def _check_train_cancel_disruption(dis):
is_valid_disruption(dis, chaos_disrup=False)
eq_(dis['disruption_id'], 'disruption_bob')
eq_(dis['severity']['effect'], 'NO_SERVICE')
eq_(len(dis['impacted_objects']), 1)
ptobj = dis['impacted_objects'][0]['pt_object']
eq_(ptobj['embedded_type'], 'trip')
eq_(ptobj['id'], 'vjA')
eq_(ptobj['name'], 'vjA')
# for cancellation we do not output the impacted stops
assert 'impacted_stops' not in dis['impacted_objects'][0]
def _check_train_cancel_disruption(dis):
is_valid_disruption(dis, chaos_disrup=False)
assert dis['disruption_id'] == 'disruption_bob'
assert dis['severity']['effect'] == 'NO_SERVICE'
assert len(dis['impacted_objects']) == 1
ptobj = dis['impacted_objects'][0]['pt_object']
assert ptobj['embedded_type'] == 'trip'
assert ptobj['id'] == 'vjA'
assert ptobj['name'] == 'vjA'
# for cancellation we do not output the impacted stops
assert 'impacted_stops' not in dis['impacted_objects'][0]
def test_vj_delay_day_after(self):
"""
send a mock kirin vj delaying on day after and test that the vj is not taken
"""
response = self.query_region(journey_basic_query +
"&data_freshness=realtime")
pt_response = self.query_region(
'vehicle_journeys/vjA?_current_datetime=20120614T1337')
# with no cancellation, we have 2 journeys, one direct and one with the vj:A:0
eq_(get_arrivals(response),
['20120614T080222', '20120614T080435']) # pt_walk + vj 08:01
eq_(get_used_vj(response), [['vjA'], []])
pt_response = self.query_region('vehicle_journeys')
eq_(len(pt_response['vehicle_journeys']), 4)
# check that we have the next vj
s_coord = "0.0000898312;0.0000898312" # coordinate of S in the dataset
r_coord = "0.00188646;0.00071865" # coordinate of R in the dataset
journey_later_query = "journeys?from={from_coord}&to={to_coord}&datetime={datetime}"\
.format(from_coord=s_coord, to_coord=r_coord, datetime="20120614T080500")
later_response = self.query_region(journey_later_query +
"&data_freshness=realtime")
eq_(get_arrivals(later_response),
['20120614T080935', '20120614T180222']) # pt_walk + vj 18:01
eq_(get_used_vj(later_response), [[], ['vjB']])
# no disruption yet
pt_response = self.query_region(
'vehicle_journeys/vjA?_current_datetime=20120614T1337')
eq_(len(pt_response['disruptions']), 0)
# sending disruption delaying VJ to the next day
self.send_mock("vjA",
"20120614",
'delayed',
[("stop_point:stopB", tstamp("20120615T070224"),
tstamp("20120615T070224")),
("stop_point:stopA", tstamp("20120615T070400"),
tstamp("20120615T070400"))],
disruption_id='96231_2015-07-28_0')
# A new vj is created
pt_response = self.query_region('vehicle_journeys')
eq_(len(pt_response['vehicle_journeys']), 5)
vj_ids = [vj['id'] for vj in pt_response['vehicle_journeys']]
assert 'vjA:modified:0:96231_2015-07-28_0' in vj_ids
# we should see the disruption
pt_response = self.query_region(
'vehicle_journeys/vjA?_current_datetime=20120614T1337')
eq_(len(pt_response['disruptions']), 1)
is_valid_disruption(pt_response['disruptions'][0], chaos_disrup=False)
eq_(pt_response['disruptions'][0]['disruption_id'],
'96231_2015-07-28_0')
new_response = self.query_region(journey_basic_query +
"&data_freshness=realtime")
eq_(get_arrivals(new_response),
['20120614T080435', '20120614T180222']) # pt_walk + vj 18:01
eq_(get_used_vj(new_response), [[], ['vjB']])
# it should not have changed anything for the theoric
new_base = self.query_region(journey_basic_query +
"&data_freshness=base_schedule")
eq_(get_arrivals(new_base), ['20120614T080222', '20120614T080435'])
eq_(get_used_vj(new_base), [['vjA'], []])
# the day after, we can use the delayed vj
journey_day_after_query = "journeys?from={from_coord}&to={to_coord}&datetime={datetime}"\
.format(from_coord=s_coord, to_coord=r_coord, datetime="20120615T070000")
day_after_response = self.query_region(journey_day_after_query +
"&data_freshness=realtime")
eq_(get_arrivals(day_after_response),
['20120615T070435', '20120615T070520']) # pt_walk + rt 07:02:24
eq_(get_used_vj(day_after_response),
[[], ['vjA:modified:0:96231_2015-07-28_0']])
# it should not have changed anything for the theoric the day after
day_after_base = self.query_region(journey_day_after_query +
"&data_freshness=base_schedule")
eq_(get_arrivals(day_after_base),
['20120615T070435', '20120615T080222'])
eq_(get_used_vj(day_after_base), [[], ['vjA']])
def test_vj_delay(self):
"""
send a mock kirin vj delay and test that the vj is not taken
"""
response = self.query_region(journey_basic_query + "&data_freshness=realtime")
# with no cancellation, we have 2 journeys, one direct and one with the vj:A:0
assert get_arrivals(response) == ['20120614T080222', '20120614T080436']
assert get_used_vj(response) == [['vjA'], []]
# we have 2 departures and 1 disruption (linked to the first passage)
departures = self.query_region("stop_points/stop_point:stopB/departures?_current_datetime=20120614T0800")
assert len(departures['disruptions']) == 1
assert len(departures['departures']) == 2
assert departures['departures'][0]['stop_date_time']['departure_date_time'] == '20120614T080100'
pt_response = self.query_region('vehicle_journeys')
assert len(pt_response['vehicle_journeys']) == 5
# no disruption yet
pt_response = self.query_region('vehicle_journeys/vjA?_current_datetime=20120614T1337')
assert len(pt_response['disruptions']) == 0
self.send_mock("vjA", "20120614", 'modified',
[UpdatedStopTime("stop_point:stopB",
arrival=tstamp("20120614T080224"), departure=tstamp("20120614T080225"),
arrival_delay=60 + 24, departure_delay=60+25,
message='cow on tracks'),
UpdatedStopTime("stop_point:stopA",
arrival=tstamp("20120614T080400"), departure=tstamp("20120614T080400"),
arrival_delay=3 * 60 + 58, departure_delay=3 * 60 + 58)],
disruption_id='vjA_delayed')
# A new vj is created
pt_response = self.query_region('vehicle_journeys')
assert len(pt_response['vehicle_journeys']) == 6
vj_ids = [vj['id'] for vj in pt_response['vehicle_journeys']]
assert 'vjA:modified:0:vjA_delayed' in vj_ids
def _check_train_delay_disruption(dis):
is_valid_disruption(dis, chaos_disrup=False)
assert dis['disruption_id'] == 'vjA_delayed'
assert dis['severity']['effect'] == 'SIGNIFICANT_DELAYS'
assert len(dis['impacted_objects']) == 1
ptobj = dis['impacted_objects'][0]['pt_object']
assert ptobj['embedded_type'] == 'trip'
assert ptobj['id'] == 'vjA'
assert ptobj['name'] == 'vjA'
# for delay we should have detail on the impacted stops
impacted_objs = get_not_null(dis['impacted_objects'][0], 'impacted_stops')
assert len(impacted_objs) == 2
imp_obj1 = impacted_objs[0]
assert get_valid_time(get_not_null(imp_obj1, 'amended_arrival_time')) == _dt(h=8, m=2, s=24)
assert get_valid_time(get_not_null(imp_obj1, 'amended_departure_time')) == _dt(h=8, m=2, s=25)
assert get_not_null(imp_obj1, 'cause') == 'cow on tracks'
assert get_not_null(imp_obj1, 'departure_status') == 'delayed'
assert get_not_null(imp_obj1, 'arrival_status') == 'delayed'
assert get_not_null(imp_obj1, 'stop_time_effect') == 'delayed'
assert get_valid_time(get_not_null(imp_obj1, 'base_arrival_time')) == _dt(8, 1, 0)
assert get_valid_time(get_not_null(imp_obj1, 'base_departure_time')) == _dt(8, 1, 0)
imp_obj2 = impacted_objs[1]
assert get_valid_time(get_not_null(imp_obj2, 'amended_arrival_time')) == _dt(h=8, m=4, s=0)
assert get_valid_time(get_not_null(imp_obj2, 'amended_departure_time')) == _dt(h=8, m=4, s=0)
assert imp_obj2['cause'] == ''
assert get_not_null(imp_obj1, 'stop_time_effect') == 'delayed'
assert get_not_null(imp_obj1, 'departure_status') == 'delayed'
assert get_not_null(imp_obj1, 'arrival_status') == 'delayed'
assert get_valid_time(get_not_null(imp_obj2, 'base_departure_time')) == _dt(8, 1, 2)
assert get_valid_time(get_not_null(imp_obj2, 'base_arrival_time')) == _dt(8, 1, 2)
# we should see the disruption
pt_response = self.query_region('vehicle_journeys/vjA?_current_datetime=20120614T1337')
assert len(pt_response['disruptions']) == 1
_check_train_delay_disruption(pt_response['disruptions'][0])
new_response = self.query_region(journey_basic_query + "&data_freshness=realtime")
assert get_arrivals(new_response) == ['20120614T080436', '20120614T080520']
assert get_used_vj(new_response) == [[], ['vjA:modified:0:vjA_delayed']]
pt_journey = new_response['journeys'][1]
check_journey(pt_journey,
Journey(sections=[Section(departure_date_time='20120614T080208', arrival_date_time='20120614T080225',
base_departure_date_time=None, base_arrival_date_time=None,
stop_date_times=[]),
Section(departure_date_time='20120614T080225', arrival_date_time='20120614T080400',
base_departure_date_time='20120614T080100', base_arrival_date_time='20120614T080102',
stop_date_times=[SectionStopDT(departure_date_time='20120614T080225',
arrival_date_time='20120614T080224',
base_departure_date_time='20120614T080100',
base_arrival_date_time='20120614T080100'),
SectionStopDT(
departure_date_time='20120614T080400',
arrival_date_time='20120614T080400',
base_departure_date_time='20120614T080102',
base_arrival_date_time='20120614T080102')]),
Section(departure_date_time='20120614T080400', arrival_date_time='20120614T080520',
base_departure_date_time=None, base_arrival_date_time=None,
stop_date_times=[])]))
# it should not have changed anything for the theoric
new_base = self.query_region(journey_basic_query + "&data_freshness=base_schedule")
assert get_arrivals(new_base) == ['20120614T080222', '20120614T080436']
assert get_used_vj(new_base), [['vjA'] == []]
# we have one delayed departure
departures = self.query_region("stop_points/stop_point:stopB/departures?_current_datetime=20120614T0800")
assert len(departures['disruptions']) == 2
assert len(departures['departures']) == 2
assert departures['departures'][0]['stop_date_time']['departure_date_time'] == '20120614T080225'
# Same as realtime except the departure date time
departures = self.query_region("stop_points/stop_point:stopB/departures?_current_datetime=20120614T0800&data_freshness=base_schedule")
assert len(departures['disruptions']) == 2
assert len(departures['departures']) == 2
assert departures['departures'][0]['stop_date_time']['departure_date_time'] == '20120614T080100'
# We send again the same disruption
self.send_mock("vjA", "20120614", 'modified',
[UpdatedStopTime("stop_point:stopB",
arrival=tstamp("20120614T080224"), departure=tstamp("20120614T080225"),
arrival_delay=60 + 24, departure_delay=60+25,
message='cow on tracks'),
UpdatedStopTime("stop_point:stopA",
arrival=tstamp("20120614T080400"), departure=tstamp("20120614T080400"),
arrival_delay=3 * 60 + 58, departure_delay=3 * 60 + 58)],
disruption_id='vjA_delayed')
# A new vj is created, but a useless vj has been cleaned, so the number of vj does not change
pt_response = self.query_region('vehicle_journeys')
assert len(pt_response['vehicle_journeys']) == 6
pt_response = self.query_region('vehicle_journeys/vjA?_current_datetime=20120614T1337')
assert len(pt_response['disruptions']) == 1
_check_train_delay_disruption(pt_response['disruptions'][0])
# so the first real-time vj created for the first disruption should be deactivated
new_response = self.query_region(journey_basic_query + "&data_freshness=realtime")
assert get_arrivals(new_response) == ['20120614T080436', '20120614T080520']
assert get_used_vj(new_response), [[] == ['vjA:modified:1:vjA_delayed']]
# it should not have changed anything for the theoric
new_base = self.query_region(journey_basic_query + "&data_freshness=base_schedule")
assert get_arrivals(new_base) == ['20120614T080222', '20120614T080436']
assert get_used_vj(new_base), [['vjA'] == []]
# we then try to send a delay on another train.
# we should not have lost the first delay
self.send_mock("vjB", "20120614", 'modified',
[UpdatedStopTime("stop_point:stopB", tstamp("20120614T180224"), tstamp("20120614T180225"),
arrival_delay=60 + 24, departure_delay=60 + 25,),
UpdatedStopTime("stop_point:stopA", tstamp("20120614T180400"), tstamp("20120614T180400"),
message="bob's in the place")])
pt_response = self.query_region('vehicle_journeys/vjA?_current_datetime=20120614T1337')
assert len(pt_response['disruptions']) == 1
_check_train_delay_disruption(pt_response['disruptions'][0])
# we should also have the disruption on vjB
assert len(self.query_region('vehicle_journeys/vjB?_current_datetime=20120614T1337')['disruptions']) == 1
###################################
# We now send a partial delete on B
###################################
self.send_mock("vjA", "20120614", 'modified',
[UpdatedStopTime("stop_point:stopB",
arrival=tstamp("20120614T080100"), departure=tstamp("20120614T080100")),
UpdatedStopTime("stop_point:stopA",
arrival=tstamp("20120614T080102"), departure=tstamp("20120614T080102"),
message='cow on tracks', arrival_skipped=True)],
disruption_id='vjA_skip_A')
# A new vj is created
vjs = self.query_region('vehicle_journeys?_current_datetime=20120614T1337')
assert len(vjs['vehicle_journeys']) == 7
vjA = self.query_region('vehicle_journeys/vjA?_current_datetime=20120614T1337')
# we now have 2 disruption on vjA
assert len(vjA['disruptions']) == 2
all_dis = {d['id']: d for d in vjA['disruptions']}
assert 'vjA_skip_A' in all_dis
dis = all_dis['vjA_skip_A']
is_valid_disruption(dis, chaos_disrup=False)
assert dis['disruption_id'] == 'vjA_skip_A'
assert dis['severity']['effect'] == 'DETOUR'
assert len(dis['impacted_objects']) == 1
ptobj = dis['impacted_objects'][0]['pt_object']
assert ptobj['embedded_type'] == 'trip'
assert ptobj['id'] == 'vjA'
assert ptobj['name'] == 'vjA'
# for delay we should have detail on the impacted stops
impacted_objs = get_not_null(dis['impacted_objects'][0], 'impacted_stops')
assert len(impacted_objs) == 2
imp_obj1 = impacted_objs[0]
assert get_valid_time(get_not_null(imp_obj1, 'amended_arrival_time')) == _dt(8, 1, 0)
assert get_valid_time(get_not_null(imp_obj1, 'amended_departure_time')) == _dt(8, 1, 0)
assert get_not_null(imp_obj1, 'stop_time_effect') == 'unchanged'
assert get_not_null(imp_obj1, 'arrival_status') == 'unchanged'
assert get_not_null(imp_obj1, 'departure_status') == 'unchanged'
assert get_valid_time(get_not_null(imp_obj1, 'base_arrival_time')) == _dt(8, 1, 0)
assert get_valid_time(get_not_null(imp_obj1, 'base_departure_time')) == _dt(8, 1, 0)
imp_obj2 = impacted_objs[1]
assert 'amended_arrival_time' not in imp_obj2
assert get_not_null(imp_obj2, 'cause') == 'cow on tracks'
assert get_not_null(imp_obj2, 'stop_time_effect') == 'deleted' # the stoptime is marked as deleted
assert get_not_null(imp_obj2, 'arrival_status') == 'deleted'
assert get_not_null(imp_obj2, 'departure_status') == 'unchanged' # the departure is not changed
assert get_valid_time(get_not_null(imp_obj2, 'base_departure_time')) == _dt(8, 1, 2)
assert get_valid_time(get_not_null(imp_obj2, 'base_arrival_time')) == _dt(8, 1, 2)
def test_vj_delay_day_after(self):
"""
send a mock kirin vj delaying on day after and test that the vj is not taken
"""
response = self.query_region(journey_basic_query +
"&data_freshness=realtime")
pt_response = self.query_region(
'vehicle_journeys/vjA?_current_datetime=20120614T1337')
# with no cancellation, we have 2 journeys, one direct and one with the vj:A:0
assert get_arrivals(response) == [
'20120614T080222', '20120614T080436'
] # pt_walk + vj 08:01
assert get_used_vj(response), [['vjA'] == []]
pt_response = self.query_region('vehicle_journeys')
initial_nb_vehicle_journeys = len(pt_response['vehicle_journeys'])
assert initial_nb_vehicle_journeys == 7
# check that we have the next vj
s_coord = "0.0000898312;0.0000898312" # coordinate of S in the dataset
r_coord = "0.00188646;0.00071865" # coordinate of R in the dataset
journey_later_query = "journeys?from={from_coord}&to={to_coord}&datetime={datetime}"\
.format(from_coord=s_coord, to_coord=r_coord, datetime="20120614T080500")
later_response = self.query_region(journey_later_query +
"&data_freshness=realtime")
assert get_arrivals(later_response) == [
'20120614T080936', '20120614T180222'
] # pt_walk + vj 18:01
assert get_used_vj(later_response), [[] == ['vjB']]
# no disruption yet
pt_response = self.query_region(
'vehicle_journeys/vjA?_current_datetime=20120614T1337')
assert len(pt_response['disruptions']) == 0
# sending disruption delaying VJ to the next day
self.send_mock(
"vjA",
"20120614",
'modified', [
UpdatedStopTime("stop_point:stopB", tstamp("20120615T070224"),
tstamp("20120615T070224")),
UpdatedStopTime("stop_point:stopA", tstamp("20120615T070400"),
tstamp("20120615T070400"))
],
disruption_id='96231_2015-07-28_0')
# A new vj is created
pt_response = self.query_region('vehicle_journeys')
assert len(
pt_response['vehicle_journeys']) == (initial_nb_vehicle_journeys +
1)
vj_ids = [vj['id'] for vj in pt_response['vehicle_journeys']]
assert 'vjA:modified:0:96231_2015-07-28_0' in vj_ids
# we should see the disruption
pt_response = self.query_region(
'vehicle_journeys/vjA?_current_datetime=20120614T1337')
assert len(pt_response['disruptions']) == 1
is_valid_disruption(pt_response['disruptions'][0], chaos_disrup=False)
assert pt_response['disruptions'][0][
'disruption_id'] == '96231_2015-07-28_0'
# In order to not disturb the test, line M which was added afterwards for shared section tests, is forbidden here
new_response = self.query_region(
journey_basic_query +
"&data_freshness=realtime&forbidden_uris[]=M&")
assert get_arrivals(new_response) == [
'20120614T080436', '20120614T180222'
] # pt_walk + vj 18:01
assert get_used_vj(new_response), [[] == ['vjB']]
# it should not have changed anything for the theoric
new_base = self.query_region(journey_basic_query +
"&data_freshness=base_schedule")
assert get_arrivals(new_base) == ['20120614T080222', '20120614T080436']
assert get_used_vj(new_base), [['vjA'] == []]
# the day after, we can use the delayed vj
journey_day_after_query = "journeys?from={from_coord}&to={to_coord}&datetime={datetime}"\
.format(from_coord=s_coord, to_coord=r_coord, datetime="20120615T070000")
day_after_response = self.query_region(journey_day_after_query +
"&data_freshness=realtime")
assert get_arrivals(day_after_response) == [
'20120615T070436', '20120615T070520'
] # pt_walk + rt 07:02:24
assert get_used_vj(day_after_response), [[] == [
'vjA:modified:0:96231_2015-07-28_0'
]]
# it should not have changed anything for the theoric the day after
day_after_base = self.query_region(journey_day_after_query +
"&data_freshness=base_schedule")
assert get_arrivals(day_after_base) == [
'20120615T070436', '20120615T080222'
]
assert get_used_vj(day_after_base), [[] == ['vjA']]
