From 157f54c16aadac157488371b0013c49cb66c2eba Mon Sep 17 00:00:00 2001 From: Jean SIMARD Date: Tue, 8 Oct 2019 09:50:49 +0200 Subject: [PATCH] Fix the unit tests for 'stop-point' object_code --- src/gtfs/read.rs | 16 ++++++++++++---- 1 file changed, 12 insertions(+), 4 deletions(-) diff --git a/src/gtfs/read.rs b/src/gtfs/read.rs index d591ab1b2..be6368132 100644 --- a/src/gtfs/read.rs +++ b/src/gtfs/read.rs @@ -1224,18 +1224,26 @@ mod tests { //validate stop_point code assert_eq!(1, stop_points.len()); let stop_point = stop_points.iter().next().unwrap().1; - assert_eq!(1, stop_point.codes.len()); - let code = stop_point.codes.iter().next().unwrap(); + assert_eq!(2, stop_point.codes.len()); + let mut codes_iterator = stop_point.codes.iter(); + let code = codes_iterator.next().unwrap(); assert_eq!("gtfs_stop_code", code.0); assert_eq!("1234", code.1); + let code = codes_iterator.next().unwrap(); + assert_eq!("source", code.0); + assert_eq!("stoppoint_id", code.1); //validate stop_area code assert_eq!(1, stop_areas.len()); let stop_area = stop_areas.iter().next().unwrap().1; - assert_eq!(1, stop_area.codes.len()); - let code = stop_area.codes.iter().next().unwrap(); + assert_eq!(2, stop_area.codes.len()); + let mut codes_iterator = stop_area.codes.iter(); + let code = codes_iterator.next().unwrap(); assert_eq!("gtfs_stop_code", code.0); assert_eq!("5678", code.1); + let code = codes_iterator.next().unwrap(); + assert_eq!("source", code.0); + assert_eq!("stoparea_id", code.1); }); }