Add basic read tests of a small pbf file

Author: jocelynj

tests/clipperton.osm.pbf

@@ -0,0 +1,75 @@
+use osmpbfreader;
+use std::fs::File;
+
+const PBF: &str = "tests/clipperton.osm.pbf";
+
+#[test]
+fn read_pbf() {
+    let file = File::open(PBF).unwrap();
+    let mut pbf = osmpbfreader::OsmPbfReader::new(file);
+
+    let mut num_nodes = 0;
+    let mut num_ways = 0;
+    let mut num_relations = 0;
+
+    for obj in pbf.iter() {
+        let obj = obj.unwrap();
+        match obj {
+            osmpbfreader::OsmObj::Node(_) => num_nodes += 1,
+            osmpbfreader::OsmObj::Way(_) => num_ways += 1,
+            osmpbfreader::OsmObj::Relation(_) => num_relations += 1,
+        }
+    }
+    assert_eq!(num_nodes, 3097);
+    assert_eq!(num_ways, 151);
+    assert_eq!(num_relations, 17);
+}
+
+#[test]
+fn read_pbf_par_iter() {
+    let file = File::open(PBF).unwrap();
+    let mut pbf = osmpbfreader::OsmPbfReader::new(file);
+
+    let mut num_nodes = 0;
+    let mut num_ways = 0;
+    let mut num_relations = 0;
+
+    for obj in pbf.par_iter() {
+        let obj = obj.unwrap();
+        match obj {
+            osmpbfreader::OsmObj::Node(_) => num_nodes += 1,
+            osmpbfreader::OsmObj::Way(_) => num_ways += 1,
+            osmpbfreader::OsmObj::Relation(_) => num_relations += 1,
+        }
+    }
+    assert_eq!(num_nodes, 3097);
+    assert_eq!(num_ways, 151);
+    assert_eq!(num_relations, 17);
+}
+
+#[test]
+fn read_pbf_highway() {
+    let file = File::open(PBF).unwrap();
+    let mut pbf = osmpbfreader::OsmPbfReader::new(file);
+
+    let mut num_nodes = 0;
+    let mut num_ways = 0;
+    let mut num_relations = 0;
+    let mut sum_ids = 0;
+
+    let objs = pbf
+        .get_objs_and_deps(|obj| obj.is_way() && obj.tags().contains_key("highway"))
+        .unwrap();
+    for (id, obj) in &objs {
+        sum_ids += id.inner_id();
+        match obj {
+            osmpbfreader::OsmObj::Node(_) => num_nodes += 1,
+            osmpbfreader::OsmObj::Way(_) => num_ways += 1,
+            osmpbfreader::OsmObj::Relation(_) => num_relations += 1,
+        }
+    }
+    assert_eq!(num_nodes, 9);
+    assert_eq!(num_ways, 3);
+    assert_eq!(num_relations, 0);
+    assert_eq!(sum_ids, 118512595213);
+}