@@ -4,7 +4,7 @@ use rand::Rng;
44
55use super :: { create_database, rng} ;
66use crate :: distance:: { DotProduct , Euclidean } ;
7- use crate :: { Database , Writer } ;
7+ use crate :: { Database , Reader , Writer } ;
88
99#[ test]
1010fn 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