Fix bug in projection

mpiannucci · Jul 25, 2023 · 8ed04c4 · 8ed04c4
1 parent 89c4698
commit 8ed04c4
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/src/utils/iter/projection.rs b/src/utils/iter/projection.rs
@@ -159,7 +159,7 @@ impl RegularCoordinateIterator {
         Self {
             start,
             step,
-            end: start + step * count as f64,
+            end: start + (step * (count - 1) as f64),
             current_index: 0,
             count,
         }
@@ -212,4 +212,15 @@ mod tests {
         assert_eq!(lats.len(), 10);
         assert_eq!(lngs.len(), 5);
     }
+
+    #[test]
+    fn test_start_end_regular_grid() {
+        let start = 0.0; 
+        let step = 0.25; 
+        let end = 359.75;
+        let count = 1440; 
+
+        let iter = super::RegularCoordinateIterator::new(start, step, count);
+        assert!((iter.end - end).abs() < f64::EPSILON);
+    }
 }