From c99b2a397a3f2ab41d9a04654279d2d7e74f2243 Mon Sep 17 00:00:00 2001 From: Joel Knighton Date: Wed, 28 Feb 2024 12:09:31 -0500 Subject: [PATCH] Fix TestVectorSimilarityFunction test. Add missing license. --- .../example/util/MultiFileDatasource.java | 16 ++++++++++++++++ .../graph/TestVectorSimilarityFunction.java | 1 + 2 files changed, 17 insertions(+) diff --git a/jvector-examples/src/main/java/io/github/jbellis/jvector/example/util/MultiFileDatasource.java b/jvector-examples/src/main/java/io/github/jbellis/jvector/example/util/MultiFileDatasource.java index 7fd421594..3f9e45097 100644 --- a/jvector-examples/src/main/java/io/github/jbellis/jvector/example/util/MultiFileDatasource.java +++ b/jvector-examples/src/main/java/io/github/jbellis/jvector/example/util/MultiFileDatasource.java @@ -1,3 +1,19 @@ +/* + * Copyright DataStax, Inc. + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + package io.github.jbellis.jvector.example.util; import io.github.jbellis.jvector.vector.VectorSimilarityFunction; diff --git a/jvector-tests/src/test/java/io/github/jbellis/jvector/graph/TestVectorSimilarityFunction.java b/jvector-tests/src/test/java/io/github/jbellis/jvector/graph/TestVectorSimilarityFunction.java index 544f72879..9ce0765a3 100644 --- a/jvector-tests/src/test/java/io/github/jbellis/jvector/graph/TestVectorSimilarityFunction.java +++ b/jvector-tests/src/test/java/io/github/jbellis/jvector/graph/TestVectorSimilarityFunction.java @@ -49,6 +49,7 @@ public void testCompareMulti() { } for (VectorSimilarityFunction vsf : VectorSimilarityFunction.values()) { + results.zero(); vsf.compareMulti(q, packedVectors, results); for (int i = 0; i < length; i++) { Assert.assertEquals(vsf.compare(q, vectors.get(ids[i])), results.get(i), 0.01f);