|
|
@@ -22,6 +22,7 @@ import org.elasticsearch.xpack.vectors.query.ScoreScriptUtils.L2Norm;
|
|
|
import org.elasticsearch.xpack.vectors.query.ScoreScriptUtils.L2NormSparse;
|
|
|
|
|
|
import java.util.Arrays;
|
|
|
+import java.util.Collections;
|
|
|
import java.util.HashMap;
|
|
|
import java.util.List;
|
|
|
import java.util.Map;
|
|
|
@@ -39,6 +40,7 @@ public class ScoreScriptUtilsTests extends ESTestCase {
|
|
|
}
|
|
|
|
|
|
private void testDenseVectorFunctions(Version indexVersion) {
|
|
|
+ String field = "vector";
|
|
|
float[] docVector = {230.0f, 300.33f, -34.8988f, 15.555f, -200.0f};
|
|
|
BytesRef encodedDocVector = mockEncodeDenseVector(docVector, indexVersion);
|
|
|
VectorScriptDocValues.DenseVectorScriptDocValues dvs = mock(VectorScriptDocValues.DenseVectorScriptDocValues.class);
|
|
|
@@ -46,65 +48,136 @@ public class ScoreScriptUtilsTests extends ESTestCase {
|
|
|
|
|
|
ScoreScript scoreScript = mock(ScoreScript.class);
|
|
|
when(scoreScript._getIndexVersion()).thenReturn(indexVersion);
|
|
|
+ when(scoreScript.getDoc()).thenReturn(Collections.singletonMap(field, dvs));
|
|
|
|
|
|
List<Number> queryVector = Arrays.asList(0.5f, 111.3f, -13.0f, 14.8f, -156.0f);
|
|
|
|
|
|
// test dotProduct
|
|
|
- DotProduct dotProduct = new DotProduct(scoreScript, queryVector);
|
|
|
- double result = dotProduct.dotProduct(dvs);
|
|
|
+ DotProduct dotProduct = new DotProduct(scoreScript, queryVector, field);
|
|
|
+ double result = dotProduct.dotProduct();
|
|
|
assertEquals("dotProduct result is not equal to the expected value!", 65425.624, result, 0.001);
|
|
|
|
|
|
// test cosineSimilarity
|
|
|
- CosineSimilarity cosineSimilarity = new CosineSimilarity(scoreScript, queryVector);
|
|
|
- double result2 = cosineSimilarity.cosineSimilarity(dvs);
|
|
|
+ CosineSimilarity cosineSimilarity = new CosineSimilarity(scoreScript, queryVector, field);
|
|
|
+ double result2 = cosineSimilarity.cosineSimilarity();
|
|
|
assertEquals("cosineSimilarity result is not equal to the expected value!", 0.790, result2, 0.001);
|
|
|
|
|
|
// test l1Norm
|
|
|
- L1Norm l1norm = new L1Norm(scoreScript, queryVector);
|
|
|
- double result3 = l1norm.l1norm(dvs);
|
|
|
+ L1Norm l1norm = new L1Norm(scoreScript, queryVector, field);
|
|
|
+ double result3 = l1norm.l1norm();
|
|
|
assertEquals("l1norm result is not equal to the expected value!", 485.184, result3, 0.001);
|
|
|
|
|
|
// test l2norm
|
|
|
- L2Norm l2norm = new L2Norm(scoreScript, queryVector);
|
|
|
- double result4 = l2norm.l2norm(dvs);
|
|
|
+ L2Norm l2norm = new L2Norm(scoreScript, queryVector, field);
|
|
|
+ double result4 = l2norm.l2norm();
|
|
|
assertEquals("l2norm result is not equal to the expected value!", 301.361, result4, 0.001);
|
|
|
|
|
|
// test dotProduct fails when queryVector has wrong number of dims
|
|
|
List<Number> invalidQueryVector = Arrays.asList(0.5, 111.3);
|
|
|
- DotProduct dotProduct2 = new DotProduct(scoreScript, invalidQueryVector);
|
|
|
- IllegalArgumentException e = expectThrows(IllegalArgumentException.class, () -> dotProduct2.dotProduct(dvs));
|
|
|
+ DotProduct dotProduct2 = new DotProduct(scoreScript, invalidQueryVector, field);
|
|
|
+ IllegalArgumentException e = expectThrows(IllegalArgumentException.class, dotProduct2::dotProduct);
|
|
|
assertThat(e.getMessage(), containsString("query vector has a different number of dimensions [2] than the document vectors [5]"));
|
|
|
|
|
|
// test cosineSimilarity fails when queryVector has wrong number of dims
|
|
|
- CosineSimilarity cosineSimilarity2 = new CosineSimilarity(scoreScript, invalidQueryVector);
|
|
|
- e = expectThrows(IllegalArgumentException.class, () -> cosineSimilarity2.cosineSimilarity(dvs));
|
|
|
+ CosineSimilarity cosineSimilarity2 = new CosineSimilarity(scoreScript, invalidQueryVector, field);
|
|
|
+ e = expectThrows(IllegalArgumentException.class, cosineSimilarity2::cosineSimilarity);
|
|
|
assertThat(e.getMessage(), containsString("query vector has a different number of dimensions [2] than the document vectors [5]"));
|
|
|
|
|
|
// test l1norm fails when queryVector has wrong number of dims
|
|
|
- L1Norm l1norm2 = new L1Norm(scoreScript, invalidQueryVector);
|
|
|
- e = expectThrows(IllegalArgumentException.class, () -> l1norm2.l1norm(dvs));
|
|
|
+ L1Norm l1norm2 = new L1Norm(scoreScript, invalidQueryVector, field);
|
|
|
+ e = expectThrows(IllegalArgumentException.class, l1norm2::l1norm);
|
|
|
assertThat(e.getMessage(), containsString("query vector has a different number of dimensions [2] than the document vectors [5]"));
|
|
|
|
|
|
// test l2norm fails when queryVector has wrong number of dims
|
|
|
- L2Norm l2norm2 = new L2Norm(scoreScript, invalidQueryVector);
|
|
|
- e = expectThrows(IllegalArgumentException.class, () -> l2norm2.l2norm(dvs));
|
|
|
+ L2Norm l2norm2 = new L2Norm(scoreScript, invalidQueryVector, field);
|
|
|
+ e = expectThrows(IllegalArgumentException.class, l2norm2::l2norm);
|
|
|
assertThat(e.getMessage(), containsString("query vector has a different number of dimensions [2] than the document vectors [5]"));
|
|
|
}
|
|
|
|
|
|
+ public void testDeprecatedDenseVectorFunctions() {
|
|
|
+ testDeprecatedDenseVectorFunctions(Version.V_7_4_0);
|
|
|
+ testDeprecatedDenseVectorFunctions(Version.CURRENT);
|
|
|
+ }
|
|
|
+
|
|
|
+ private void testDeprecatedDenseVectorFunctions(Version indexVersion) {
|
|
|
+ float[] docVector = {230.0f, 300.33f, -34.8988f, 15.555f, -200.0f};
|
|
|
+ BytesRef encodedDocVector = mockEncodeDenseVector(docVector, indexVersion);
|
|
|
+ VectorScriptDocValues.DenseVectorScriptDocValues dvs = mock(VectorScriptDocValues.DenseVectorScriptDocValues.class);
|
|
|
+ when(dvs.getEncodedValue()).thenReturn(encodedDocVector);
|
|
|
+
|
|
|
+ ScoreScript scoreScript = mock(ScoreScript.class);
|
|
|
+ when(scoreScript._getIndexVersion()).thenReturn(indexVersion);
|
|
|
+
|
|
|
+ List<Number> queryVector = Arrays.asList(0.5f, 111.3f, -13.0f, 14.8f, -156.0f);
|
|
|
+
|
|
|
+ // test dotProduct
|
|
|
+ DotProduct dotProduct = new DotProduct(scoreScript, queryVector, dvs);
|
|
|
+ double result = dotProduct.dotProduct();
|
|
|
+ assertEquals("dotProduct result is not equal to the expected value!", 65425.624, result, 0.001);
|
|
|
+ assertWarnings(ScoreScriptUtils.DEPRECATION_MESSAGE);
|
|
|
+
|
|
|
+ // test cosineSimilarity
|
|
|
+ CosineSimilarity cosineSimilarity = new CosineSimilarity(scoreScript, queryVector, dvs);
|
|
|
+ double result2 = cosineSimilarity.cosineSimilarity();
|
|
|
+ assertEquals("cosineSimilarity result is not equal to the expected value!", 0.790, result2, 0.001);
|
|
|
+ assertWarnings(ScoreScriptUtils.DEPRECATION_MESSAGE);
|
|
|
+
|
|
|
+ // test l1Norm
|
|
|
+ L1Norm l1norm = new L1Norm(scoreScript, queryVector, dvs);
|
|
|
+ double result3 = l1norm.l1norm();
|
|
|
+ assertEquals("l1norm result is not equal to the expected value!", 485.184, result3, 0.001);
|
|
|
+ assertWarnings(ScoreScriptUtils.DEPRECATION_MESSAGE);
|
|
|
+
|
|
|
+ // test l2norm
|
|
|
+ L2Norm l2norm = new L2Norm(scoreScript, queryVector, dvs);
|
|
|
+ double result4 = l2norm.l2norm();
|
|
|
+ assertEquals("l2norm result is not equal to the expected value!", 301.361, result4, 0.001);
|
|
|
+ assertWarnings(ScoreScriptUtils.DEPRECATION_MESSAGE);
|
|
|
+
|
|
|
+ // test dotProduct fails when queryVector has wrong number of dims
|
|
|
+ List<Number> invalidQueryVector = Arrays.asList(0.5, 111.3);
|
|
|
+ DotProduct dotProduct2 = new DotProduct(scoreScript, invalidQueryVector, dvs);
|
|
|
+ IllegalArgumentException e = expectThrows(IllegalArgumentException.class, dotProduct2::dotProduct);
|
|
|
+ assertThat(e.getMessage(), containsString("query vector has a different number of dimensions [2] than the document vectors [5]"));
|
|
|
+ assertWarnings(ScoreScriptUtils.DEPRECATION_MESSAGE);
|
|
|
+
|
|
|
+ // test cosineSimilarity fails when queryVector has wrong number of dims
|
|
|
+ CosineSimilarity cosineSimilarity2 = new CosineSimilarity(scoreScript, invalidQueryVector, dvs);
|
|
|
+ e = expectThrows(IllegalArgumentException.class, cosineSimilarity2::cosineSimilarity);
|
|
|
+ assertThat(e.getMessage(), containsString("query vector has a different number of dimensions [2] than the document vectors [5]"));
|
|
|
+ assertWarnings(ScoreScriptUtils.DEPRECATION_MESSAGE);
|
|
|
+
|
|
|
+ // test l1norm fails when queryVector has wrong number of dims
|
|
|
+ L1Norm l1norm2 = new L1Norm(scoreScript, invalidQueryVector, dvs);
|
|
|
+ e = expectThrows(IllegalArgumentException.class, l1norm2::l1norm);
|
|
|
+ assertThat(e.getMessage(), containsString("query vector has a different number of dimensions [2] than the document vectors [5]"));
|
|
|
+ assertWarnings(ScoreScriptUtils.DEPRECATION_MESSAGE);
|
|
|
+
|
|
|
+ // test l2norm fails when queryVector has wrong number of dims
|
|
|
+ L2Norm l2norm2 = new L2Norm(scoreScript, invalidQueryVector, dvs);
|
|
|
+ e = expectThrows(IllegalArgumentException.class, l2norm2::l2norm);
|
|
|
+ assertThat(e.getMessage(), containsString("query vector has a different number of dimensions [2] than the document vectors [5]"));
|
|
|
+ assertWarnings(ScoreScriptUtils.DEPRECATION_MESSAGE);
|
|
|
+ }
|
|
|
+
|
|
|
public void testSparseVectorFunctions() {
|
|
|
testSparseVectorFunctions(Version.V_7_4_0);
|
|
|
testSparseVectorFunctions(Version.CURRENT);
|
|
|
}
|
|
|
|
|
|
private void testSparseVectorFunctions(Version indexVersion) {
|
|
|
+ String field = "vector";
|
|
|
+
|
|
|
int[] docVectorDims = {2, 10, 50, 113, 4545};
|
|
|
float[] docVectorValues = {230.0f, 300.33f, -34.8988f, 15.555f, -200.0f};
|
|
|
BytesRef encodedDocVector = VectorEncoderDecoder.encodeSparseVector(
|
|
|
indexVersion, docVectorDims, docVectorValues, docVectorDims.length);
|
|
|
VectorScriptDocValues.SparseVectorScriptDocValues dvs = mock(VectorScriptDocValues.SparseVectorScriptDocValues.class);
|
|
|
when(dvs.getEncodedValue()).thenReturn(encodedDocVector);
|
|
|
+
|
|
|
ScoreScript scoreScript = mock(ScoreScript.class);
|
|
|
when(scoreScript._getIndexVersion()).thenReturn(indexVersion);
|
|
|
+ when(scoreScript.getDoc()).thenReturn(Collections.singletonMap(field, dvs));
|
|
|
|
|
|
Map<String, Number> queryVector = new HashMap<String, Number>() {{
|
|
|
put("2", 0.5);
|
|
|
@@ -115,29 +188,79 @@ public class ScoreScriptUtilsTests extends ESTestCase {
|
|
|
}};
|
|
|
|
|
|
// test dotProduct
|
|
|
- DotProductSparse docProductSparse = new DotProductSparse(scoreScript, queryVector);
|
|
|
- double result = docProductSparse.dotProductSparse(dvs);
|
|
|
+ DotProductSparse docProductSparse = new DotProductSparse(scoreScript, queryVector, field);
|
|
|
+ double result = docProductSparse.dotProductSparse();
|
|
|
assertEquals("dotProductSparse result is not equal to the expected value!", 65425.624, result, 0.001);
|
|
|
|
|
|
// test cosineSimilarity
|
|
|
- CosineSimilaritySparse cosineSimilaritySparse = new CosineSimilaritySparse(scoreScript, queryVector);
|
|
|
- double result2 = cosineSimilaritySparse.cosineSimilaritySparse(dvs);
|
|
|
+ CosineSimilaritySparse cosineSimilaritySparse = new CosineSimilaritySparse(scoreScript, queryVector, field);
|
|
|
+ double result2 = cosineSimilaritySparse.cosineSimilaritySparse();
|
|
|
assertEquals("cosineSimilaritySparse result is not equal to the expected value!", 0.790, result2, 0.001);
|
|
|
|
|
|
// test l1norm
|
|
|
- L1NormSparse l1Norm = new L1NormSparse(scoreScript, queryVector);
|
|
|
- double result3 = l1Norm.l1normSparse(dvs);
|
|
|
+ L1NormSparse l1Norm = new L1NormSparse(scoreScript, queryVector, field);
|
|
|
+ double result3 = l1Norm.l1normSparse();
|
|
|
assertEquals("l1normSparse result is not equal to the expected value!", 485.184, result3, 0.001);
|
|
|
|
|
|
// test l2norm
|
|
|
- L2NormSparse l2Norm = new L2NormSparse(scoreScript, queryVector);
|
|
|
- double result4 = l2Norm.l2normSparse(dvs);
|
|
|
+ L2NormSparse l2Norm = new L2NormSparse(scoreScript, queryVector, field);
|
|
|
+ double result4 = l2Norm.l2normSparse();
|
|
|
assertEquals("l2normSparse result is not equal to the expected value!", 301.361, result4, 0.001);
|
|
|
-
|
|
|
assertWarnings(SparseVectorFieldMapper.DEPRECATION_MESSAGE);
|
|
|
}
|
|
|
|
|
|
+ public void testDeprecatedSparseVectorFunctions() {
|
|
|
+ testDeprecatedSparseVectorFunctions(Version.V_7_4_0);
|
|
|
+ testDeprecatedSparseVectorFunctions(Version.CURRENT);
|
|
|
+ }
|
|
|
+
|
|
|
+ private void testDeprecatedSparseVectorFunctions(Version indexVersion) {
|
|
|
+ int[] docVectorDims = {2, 10, 50, 113, 4545};
|
|
|
+ float[] docVectorValues = {230.0f, 300.33f, -34.8988f, 15.555f, -200.0f};
|
|
|
+ BytesRef encodedDocVector = VectorEncoderDecoder.encodeSparseVector(
|
|
|
+ indexVersion, docVectorDims, docVectorValues, docVectorDims.length);
|
|
|
+ VectorScriptDocValues.SparseVectorScriptDocValues dvs = mock(VectorScriptDocValues.SparseVectorScriptDocValues.class);
|
|
|
+ when(dvs.getEncodedValue()).thenReturn(encodedDocVector);
|
|
|
+
|
|
|
+ ScoreScript scoreScript = mock(ScoreScript.class);
|
|
|
+ when(scoreScript._getIndexVersion()).thenReturn(indexVersion);
|
|
|
+
|
|
|
+ Map<String, Number> queryVector = new HashMap<String, Number>() {{
|
|
|
+ put("2", 0.5);
|
|
|
+ put("10", 111.3);
|
|
|
+ put("50", -13.0);
|
|
|
+ put("113", 14.8);
|
|
|
+ put("4545", -156.0);
|
|
|
+ }};
|
|
|
+
|
|
|
+ // test dotProduct
|
|
|
+ DotProductSparse docProductSparse = new DotProductSparse(scoreScript, queryVector, dvs);
|
|
|
+ double result = docProductSparse.dotProductSparse();
|
|
|
+ assertEquals("dotProductSparse result is not equal to the expected value!", 65425.624, result, 0.001);
|
|
|
+ assertWarnings(SparseVectorFieldMapper.DEPRECATION_MESSAGE, ScoreScriptUtils.DEPRECATION_MESSAGE);
|
|
|
+
|
|
|
+ // test cosineSimilarity
|
|
|
+ CosineSimilaritySparse cosineSimilaritySparse = new CosineSimilaritySparse(scoreScript, queryVector, dvs);
|
|
|
+ double result2 = cosineSimilaritySparse.cosineSimilaritySparse();
|
|
|
+ assertEquals("cosineSimilaritySparse result is not equal to the expected value!", 0.790, result2, 0.001);
|
|
|
+ assertWarnings(SparseVectorFieldMapper.DEPRECATION_MESSAGE, ScoreScriptUtils.DEPRECATION_MESSAGE);
|
|
|
+
|
|
|
+ // test l1norm
|
|
|
+ L1NormSparse l1Norm = new L1NormSparse(scoreScript, queryVector, dvs);
|
|
|
+ double result3 = l1Norm.l1normSparse();
|
|
|
+ assertEquals("l1normSparse result is not equal to the expected value!", 485.184, result3, 0.001);
|
|
|
+ assertWarnings(SparseVectorFieldMapper.DEPRECATION_MESSAGE, ScoreScriptUtils.DEPRECATION_MESSAGE);
|
|
|
+
|
|
|
+ // test l2norm
|
|
|
+ L2NormSparse l2Norm = new L2NormSparse(scoreScript, queryVector, dvs);
|
|
|
+ double result4 = l2Norm.l2normSparse();
|
|
|
+ assertEquals("l2normSparse result is not equal to the expected value!", 301.361, result4, 0.001);
|
|
|
+ assertWarnings(SparseVectorFieldMapper.DEPRECATION_MESSAGE, ScoreScriptUtils.DEPRECATION_MESSAGE);
|
|
|
+ }
|
|
|
+
|
|
|
public void testSparseVectorMissingDimensions1() {
|
|
|
+ String field = "vector";
|
|
|
+
|
|
|
// Document vector's biggest dimension > query vector's biggest dimension
|
|
|
int[] docVectorDims = {2, 10, 50, 113, 4545, 4546};
|
|
|
float[] docVectorValues = {230.0f, 300.33f, -34.8988f, 15.555f, -200.0f, 11.5f};
|
|
|
@@ -145,8 +268,11 @@ public class ScoreScriptUtilsTests extends ESTestCase {
|
|
|
Version.CURRENT, docVectorDims, docVectorValues, docVectorDims.length);
|
|
|
VectorScriptDocValues.SparseVectorScriptDocValues dvs = mock(VectorScriptDocValues.SparseVectorScriptDocValues.class);
|
|
|
when(dvs.getEncodedValue()).thenReturn(encodedDocVector);
|
|
|
+
|
|
|
ScoreScript scoreScript = mock(ScoreScript.class);
|
|
|
when(scoreScript._getIndexVersion()).thenReturn(Version.CURRENT);
|
|
|
+ when(scoreScript.getDoc()).thenReturn(Collections.singletonMap(field, dvs));
|
|
|
+
|
|
|
Map<String, Number> queryVector = new HashMap<String, Number>() {{
|
|
|
put("2", 0.5);
|
|
|
put("10", 111.3);
|
|
|
@@ -157,29 +283,33 @@ public class ScoreScriptUtilsTests extends ESTestCase {
|
|
|
}};
|
|
|
|
|
|
// test dotProduct
|
|
|
- DotProductSparse docProductSparse = new DotProductSparse(scoreScript, queryVector);
|
|
|
- double result = docProductSparse.dotProductSparse(dvs);
|
|
|
+ DotProductSparse docProductSparse = new DotProductSparse(scoreScript, queryVector, field);
|
|
|
+ double result = docProductSparse.dotProductSparse();
|
|
|
assertEquals("dotProductSparse result is not equal to the expected value!", 65425.624, result, 0.001);
|
|
|
+ assertWarnings(SparseVectorFieldMapper.DEPRECATION_MESSAGE);
|
|
|
|
|
|
// test cosineSimilarity
|
|
|
- CosineSimilaritySparse cosineSimilaritySparse = new CosineSimilaritySparse(scoreScript, queryVector);
|
|
|
- double result2 = cosineSimilaritySparse.cosineSimilaritySparse(dvs);
|
|
|
+ CosineSimilaritySparse cosineSimilaritySparse = new CosineSimilaritySparse(scoreScript, queryVector, field);
|
|
|
+ double result2 = cosineSimilaritySparse.cosineSimilaritySparse();
|
|
|
assertEquals("cosineSimilaritySparse result is not equal to the expected value!", 0.786, result2, 0.001);
|
|
|
+ assertWarnings(SparseVectorFieldMapper.DEPRECATION_MESSAGE);
|
|
|
|
|
|
// test l1norm
|
|
|
- L1NormSparse l1Norm = new L1NormSparse(scoreScript, queryVector);
|
|
|
- double result3 = l1Norm.l1normSparse(dvs);
|
|
|
+ L1NormSparse l1Norm = new L1NormSparse(scoreScript, queryVector, field);
|
|
|
+ double result3 = l1Norm.l1normSparse();
|
|
|
assertEquals("l1normSparse result is not equal to the expected value!", 517.184, result3, 0.001);
|
|
|
+ assertWarnings(SparseVectorFieldMapper.DEPRECATION_MESSAGE);
|
|
|
|
|
|
// test l2norm
|
|
|
- L2NormSparse l2Norm = new L2NormSparse(scoreScript, queryVector);
|
|
|
- double result4 = l2Norm.l2normSparse(dvs);
|
|
|
+ L2NormSparse l2Norm = new L2NormSparse(scoreScript, queryVector, field);
|
|
|
+ double result4 = l2Norm.l2normSparse();
|
|
|
assertEquals("l2normSparse result is not equal to the expected value!", 302.277, result4, 0.001);
|
|
|
-
|
|
|
assertWarnings(SparseVectorFieldMapper.DEPRECATION_MESSAGE);
|
|
|
}
|
|
|
|
|
|
public void testSparseVectorMissingDimensions2() {
|
|
|
+ String field = "vector";
|
|
|
+
|
|
|
// Document vector's biggest dimension < query vector's biggest dimension
|
|
|
int[] docVectorDims = {2, 10, 50, 113, 4545, 4546};
|
|
|
float[] docVectorValues = {230.0f, 300.33f, -34.8988f, 15.555f, -200.0f, 11.5f};
|
|
|
@@ -187,8 +317,11 @@ public class ScoreScriptUtilsTests extends ESTestCase {
|
|
|
Version.CURRENT, docVectorDims, docVectorValues, docVectorDims.length);
|
|
|
VectorScriptDocValues.SparseVectorScriptDocValues dvs = mock(VectorScriptDocValues.SparseVectorScriptDocValues.class);
|
|
|
when(dvs.getEncodedValue()).thenReturn(encodedDocVector);
|
|
|
+
|
|
|
ScoreScript scoreScript = mock(ScoreScript.class);
|
|
|
when(scoreScript._getIndexVersion()).thenReturn(Version.CURRENT);
|
|
|
+ when(scoreScript.getDoc()).thenReturn(Collections.singletonMap(field, dvs));
|
|
|
+
|
|
|
Map<String, Number> queryVector = new HashMap<String, Number>() {{
|
|
|
put("2", 0.5);
|
|
|
put("10", 111.3);
|
|
|
@@ -199,25 +332,27 @@ public class ScoreScriptUtilsTests extends ESTestCase {
|
|
|
}};
|
|
|
|
|
|
// test dotProduct
|
|
|
- DotProductSparse docProductSparse = new DotProductSparse(scoreScript, queryVector);
|
|
|
- double result = docProductSparse.dotProductSparse(dvs);
|
|
|
+ DotProductSparse docProductSparse = new DotProductSparse(scoreScript, queryVector, field);
|
|
|
+ double result = docProductSparse.dotProductSparse();
|
|
|
assertEquals("dotProductSparse result is not equal to the expected value!", 65425.624, result, 0.001);
|
|
|
+ assertWarnings(SparseVectorFieldMapper.DEPRECATION_MESSAGE);
|
|
|
|
|
|
// test cosineSimilarity
|
|
|
- CosineSimilaritySparse cosineSimilaritySparse = new CosineSimilaritySparse(scoreScript, queryVector);
|
|
|
- double result2 = cosineSimilaritySparse.cosineSimilaritySparse(dvs);
|
|
|
+ CosineSimilaritySparse cosineSimilaritySparse = new CosineSimilaritySparse(scoreScript, queryVector, field);
|
|
|
+ double result2 = cosineSimilaritySparse.cosineSimilaritySparse();
|
|
|
assertEquals("cosineSimilaritySparse result is not equal to the expected value!", 0.786, result2, 0.001);
|
|
|
+ assertWarnings(SparseVectorFieldMapper.DEPRECATION_MESSAGE);
|
|
|
|
|
|
// test l1norm
|
|
|
- L1NormSparse l1Norm = new L1NormSparse(scoreScript, queryVector);
|
|
|
- double result3 = l1Norm.l1normSparse(dvs);
|
|
|
+ L1NormSparse l1Norm = new L1NormSparse(scoreScript, queryVector, field);
|
|
|
+ double result3 = l1Norm.l1normSparse();
|
|
|
assertEquals("l1normSparse result is not equal to the expected value!", 517.184, result3, 0.001);
|
|
|
+ assertWarnings(SparseVectorFieldMapper.DEPRECATION_MESSAGE);
|
|
|
|
|
|
// test l2norm
|
|
|
- L2NormSparse l2Norm = new L2NormSparse(scoreScript, queryVector);
|
|
|
- double result4 = l2Norm.l2normSparse(dvs);
|
|
|
+ L2NormSparse l2Norm = new L2NormSparse(scoreScript, queryVector, field);
|
|
|
+ double result4 = l2Norm.l2normSparse();
|
|
|
assertEquals("l2normSparse result is not equal to the expected value!", 302.277, result4, 0.001);
|
|
|
-
|
|
|
assertWarnings(SparseVectorFieldMapper.DEPRECATION_MESSAGE);
|
|
|
}
|
|
|
}
|
Julie Tibshirani