Test same values with euclidean_distance and dot_product functions

trinodb · Jul 24, 2024 · 0002afa · 0002afa
1 parent 8f38e96
commit 0002afa
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/core/trino-main/src/test/java/io/trino/operator/scalar/TestArrayVectorFunctions.java b/core/trino-main/src/test/java/io/trino/operator/scalar/TestArrayVectorFunctions.java
@@ -49,6 +49,9 @@ void testEuclideanDistance()
         assertThat(assertions.function("euclidean_distance", "ARRAY[1, 2]", "ARRAY[3, 4]"))
                 .hasType(DOUBLE)
                 .isEqualTo(2.8284271247461903);
+        assertThat(assertions.function("euclidean_distance", "ARRAY[4, 5, 6]", "ARRAY[4, 5, 6]"))
+                .hasType(DOUBLE)
+                .isEqualTo(0.0);
         assertThat(assertions.function("euclidean_distance", "ARRAY[REAL '1.1', REAL '2.2', REAL '3.3']", "ARRAY[REAL '4.4', REAL '5.5', REAL '6.6']"))
                 .hasType(DOUBLE)
                 .isEqualTo(5.715767651212193);
@@ -147,6 +150,9 @@ void testDotProduct()
         assertThat(assertions.function("dot_product", "ARRAY[1, 2]", "ARRAY[3, 4]"))
                 .hasType(DOUBLE)
                 .isEqualTo(11.0);
+        assertThat(assertions.function("dot_product", "ARRAY[4, 5, 6]", "ARRAY[4, 5, 6]"))
+                .hasType(DOUBLE)
+                .isEqualTo(77.0);
         assertThat(assertions.function("dot_product", "ARRAY[REAL '1.1', REAL '2.2', REAL '3.3']", "ARRAY[REAL '4.4', REAL '5.5', REAL '6.6']"))
                 .hasType(DOUBLE)
                 .isEqualTo(38.719999842643745);