/*
* Copyright (c) Microsoft Corporation.
* Licensed under the MIT license.
*/
use diskann_quantization::multi_vector::{Mat, MatMut, Standard};
// Test that `rows` on MatMut correctly captures an immutable borrow,
// preventing mutation of the MatMut while the iterator is in scope.
fn main() {
let mut mat: Mat<Standard<f32>> = Mat::new(Standard::new(4, 3).unwrap(), 0.0f32).unwrap();
let mut view: MatMut<'_, Standard<f32>> = mat.as_view_mut();
let iter = view.rows();
// This should fail: we cannot mutably borrow `view` while `iter` exists
let _ = view.get_row_mut(0);
for row in iter {
assert_eq!(row[0], 0.0);
}
}