@@ -621,24 +621,24 @@ mod tests {
621621 fn test_read_fasttext ( ) {
622622 let embeddings = read_fasttext ( ) ;
623623 let results = embeddings. word_similarity ( "über" , 3 ) . unwrap ( ) ;
624- assert_eq ! ( results[ 0 ] . word, "auf" ) ;
625- assert_abs_diff_eq ! ( * results[ 0 ] . similarity , 0.568513 , epsilon = 1e-6 ) ;
626- assert_eq ! ( results[ 1 ] . word, "vor" ) ;
627- assert_abs_diff_eq ! ( * results[ 1 ] . similarity , 0.551551 , epsilon = 1e-6 ) ;
628- assert_eq ! ( results[ 2 ] . word, "durch" ) ;
629- assert_abs_diff_eq ! ( * results[ 2 ] . similarity , 0.547349 , epsilon = 1e-6 ) ;
624+ assert_eq ! ( results[ 0 ] . word( ) , "auf" ) ;
625+ assert_abs_diff_eq ! ( results[ 0 ] . cosine_similarity ( ) , 0.568513 , epsilon = 1e-6 ) ;
626+ assert_eq ! ( results[ 1 ] . word( ) , "vor" ) ;
627+ assert_abs_diff_eq ! ( results[ 1 ] . cosine_similarity ( ) , 0.551551 , epsilon = 1e-6 ) ;
628+ assert_eq ! ( results[ 2 ] . word( ) , "durch" ) ;
629+ assert_abs_diff_eq ! ( results[ 2 ] . cosine_similarity ( ) , 0.547349 , epsilon = 1e-6 ) ;
630630 }
631631
632632 #[ test]
633633 fn test_read_fasttext_unknown ( ) {
634634 let embeddings = read_fasttext ( ) ;
635635 let results = embeddings. word_similarity ( "unknown" , 3 ) . unwrap ( ) ;
636- assert_eq ! ( results[ 0 ] . word, "einer" ) ;
637- assert_abs_diff_eq ! ( * results[ 0 ] . similarity , 0.691177 , epsilon = 1e-6 ) ;
638- assert_eq ! ( results[ 1 ] . word, "und" ) ;
639- assert_abs_diff_eq ! ( * results[ 1 ] . similarity , 0.576449 , epsilon = 1e-6 ) ;
640- assert_eq ! ( results[ 2 ] . word, "des" ) ;
641- assert_abs_diff_eq ! ( * results[ 2 ] . similarity , 0.570398 , epsilon = 1e-6 ) ;
636+ assert_eq ! ( results[ 0 ] . word( ) , "einer" ) ;
637+ assert_abs_diff_eq ! ( results[ 0 ] . cosine_similarity ( ) , 0.691177 , epsilon = 1e-6 ) ;
638+ assert_eq ! ( results[ 1 ] . word( ) , "und" ) ;
639+ assert_abs_diff_eq ! ( results[ 1 ] . cosine_similarity ( ) , 0.576449 , epsilon = 1e-6 ) ;
640+ assert_eq ! ( results[ 2 ] . word( ) , "des" ) ;
641+ assert_abs_diff_eq ! ( results[ 2 ] . cosine_similarity ( ) , 0.570398 , epsilon = 1e-6 ) ;
642642 }
643643
644644 #[ test]
0 commit comments