Skip to content

Commit 54fa7ae

Browse files
committed
Improve the test to check that we do not delete more than necessary
1 parent ab45cc4 commit 54fa7ae

File tree

1 file changed

+15
-7
lines changed

1 file changed

+15
-7
lines changed

src/tests/writer.rs

Lines changed: 15 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@ use rand::Rng;
44

55
use super::{create_database, rng};
66
use crate::distance::{DotProduct, Euclidean};
7-
use crate::{Database, Writer};
7+
use crate::{Database, Reader, Writer};
88

99
#[test]
1010
fn clear_small_database() {
@@ -15,15 +15,23 @@ fn clear_small_database() {
1515

1616
let mut wtxn = env.write_txn().unwrap();
1717
let database: Database<DotProduct> = env.create_database(&mut wtxn, None).unwrap();
18-
let writer = Writer::new(database, 0, 3);
19-
writer.add_item(&mut wtxn, 0, &[0.0, 1.0, 2.0]).unwrap();
20-
writer.clear(&mut wtxn).unwrap();
21-
writer.build(&mut wtxn, &mut rng(), None).unwrap();
18+
19+
let zero_writer = Writer::new(database, 0, 3);
20+
zero_writer.add_item(&mut wtxn, 0, &[0.0, 1.0, 2.0]).unwrap();
21+
zero_writer.clear(&mut wtxn).unwrap();
22+
zero_writer.build(&mut wtxn, &mut rng(), None).unwrap();
23+
24+
let one_writer = Writer::new(database, 1, 3);
25+
one_writer.add_item(&mut wtxn, 0, &[1.0, 2.0, 3.0]).unwrap();
26+
one_writer.build(&mut wtxn, &mut rng(), None).unwrap();
2227
wtxn.commit().unwrap();
2328

2429
let mut wtxn = env.write_txn().unwrap();
25-
let writer = Writer::new(database, 0, 3);
26-
writer.clear(&mut wtxn).unwrap();
30+
let zero_writer = Writer::new(database, 0, 3);
31+
zero_writer.clear(&mut wtxn).unwrap();
32+
33+
let one_reader = Reader::open(&wtxn, 1, database).unwrap();
34+
assert_eq!(one_reader.item_vector(&wtxn, 0).unwrap().unwrap(), vec![1.0, 2.0, 3.0]);
2735
wtxn.commit().unwrap();
2836
}
2937

0 commit comments

Comments
 (0)