Test Ball::contains.

finnbear · Jan 7, 2025 · bb2d0ae · bb2d0ae
1 parent 38e4143
commit bb2d0ae
Showing 1 changed file with 15 additions and 1 deletion.
diff --git a/src/ball.rs b/src/ball.rs
@@ -37,7 +37,7 @@ impl<T: BHValue, const D: usize> Ball<T, D> {
         for i in 0..D {
             distance_squared += (point[i] - self.center[i]).powi(2);
         }
-        distance_squared.sqrt()
+        distance_squared <= self.radius.powi(2)
     }
 }
 
@@ -57,3 +57,17 @@ impl<T: BHValue, const D: usize> IntersectsAabb<T, D> for Ball<T, D> {
         distance_squared <= self.radius.powi(2)
     }
 }
+
+#[cfg(test)]
+mod tests {
+    use nalgebra::Point;
+
+    use super::Ball;
+
+    #[test]
+    fn ball_contains() {
+        let ball = Ball::new(Point::<f32, 3>::new(3.0, 4.0, 5.0), 2.0);
+        assert!(ball.contains(&Point::<f32, 3>::new(4.0, 4.0, 5.0)));
+        assert!(!ball.contains(&Point::<f32, 3>::new(-4.0, 4.0, 5.0)));
+    }
+}