Skip to content

Commit d33dadb

Browse files
add coverage test on euclidean distance
1 parent 678a5e1 commit d33dadb

1 file changed

Lines changed: 12 additions & 0 deletions

File tree

src/machine_learning/k_nearest_neighbors.rs

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -111,4 +111,16 @@ mod tests {
111111
let result = k_nearest_neighbors(training_data, test_point, 10);
112112
assert_eq!(result, None);
113113
}
114+
115+
#[test]
116+
fn test_euclidean_distance_different_dimensions() {
117+
let training_data = vec![
118+
(vec![1.0, 2.0], 0.0),
119+
(vec![2.0, 3.0], 0.0),
120+
(vec![5.0], 1.0),
121+
];
122+
let test_point = vec![1.5, 2.5];
123+
let result = k_nearest_neighbors(training_data, test_point, 2);
124+
assert_eq!(result, Some(0.0));
125+
}
114126
}

0 commit comments

Comments
 (0)