Skip to content

Commit c361560

Browse files
committed
test(pipnn): name zero-k fixture inputs
1 parent 70b80c0 commit c361560

1 file changed

Lines changed: 10 additions & 2 deletions

File tree

diskann/src/graph/pipnn/leaf_build.rs

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -642,13 +642,21 @@ mod tests {
642642
fn zero_k_adds_no_candidates() {
643643
// Given
644644
let point_values = [0.0_f32, 1.0, 2.0];
645+
let point_count = 3;
646+
let dimensions = 1;
647+
let zero_k = 0;
645648
let leaves = [vec![0, 1, 2]];
646649
let expected_adjacency: [Vec<u32>; 3] = [vec![], vec![], vec![]];
647650

648651
// When
649652
let actual_adjacency = adjacency_lists(
650-
build_candidate_graph(matrix_view(&point_values, 3, 1), &leaves, 0, Metric::L2)
651-
.unwrap(),
653+
build_candidate_graph(
654+
matrix_view(&point_values, point_count, dimensions),
655+
&leaves,
656+
zero_k,
657+
Metric::L2,
658+
)
659+
.unwrap(),
652660
);
653661

654662
// Then

0 commit comments

Comments
 (0)