From 81f004c84569a5b2ca2b6a30a3b729966ca70777 Mon Sep 17 00:00:00 2001 From: Mark Wolters Date: Fri, 18 Aug 2023 11:33:01 -0400 Subject: [PATCH] removing pinecone adapter test class --- .../pinecone/PineconeOpMapperTest.java | 227 ------------------ 1 file changed, 227 deletions(-) delete mode 100644 adapter-pinecone/src/test/java/io/nosqlbench/adapter/pinecone/PineconeOpMapperTest.java diff --git a/adapter-pinecone/src/test/java/io/nosqlbench/adapter/pinecone/PineconeOpMapperTest.java b/adapter-pinecone/src/test/java/io/nosqlbench/adapter/pinecone/PineconeOpMapperTest.java deleted file mode 100644 index 14b250c4b..000000000 --- a/adapter-pinecone/src/test/java/io/nosqlbench/adapter/pinecone/PineconeOpMapperTest.java +++ /dev/null @@ -1,227 +0,0 @@ -/* - * Copyright (c) 2023 nosqlbench - * - * 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.nosqlbench.adapter.pinecone; - -import io.nosqlbench.adapter.pinecone.opdispensers.*; -import io.nosqlbench.adapter.pinecone.ops.*; -import io.nosqlbench.api.config.NBLabeledElement; -import io.nosqlbench.api.config.standard.NBConfiguration; -import io.nosqlbench.adapters.api.activityconfig.OpsLoader; -import io.nosqlbench.adapters.api.activityconfig.yaml.OpTemplate; -import io.nosqlbench.adapters.api.activityconfig.yaml.OpTemplateFormat; -import io.nosqlbench.adapters.api.activityconfig.yaml.OpsDocList; -import io.nosqlbench.adapters.api.activityimpl.OpDispenser; -import io.nosqlbench.adapters.api.activityimpl.uniform.DriverSpaceCache; -import io.nosqlbench.adapters.api.templating.ParsedOp; -import org.apache.logging.log4j.LogManager; -import org.apache.logging.log4j.Logger; -import org.junit.jupiter.api.BeforeAll; -import org.junit.jupiter.api.Test; - -import java.util.List; -import java.util.Map; - -public class PineconeOpMapperTest { - - private final static Logger logger = LogManager.getLogger(PineconeOpMapperTest.class); - static NBConfiguration cfg; - static PineconeDriverAdapter adapter; - static PineconeOpMapper mapper; - - @BeforeAll - public static void initializeTestMapper() { - Map configMap = Map.of("apiKey","2f55b2f0-670f-4c51-9073-4d37142b761a", - "environment","us-east-1-aws", - "projectName","default"); - cfg = PineconeSpace.getConfigModel().apply(configMap); - adapter = new PineconeDriverAdapter(); - adapter.applyConfig(cfg); - DriverSpaceCache cache = adapter.getSpaceCache(); - mapper = new PineconeOpMapper(adapter, cache, cfg); - } - - private static ParsedOp parsedOpFor(String yaml) { - OpsDocList docs = OpsLoader.loadString(yaml, OpTemplateFormat.yaml, Map.of(), null); - OpTemplate opTemplate = docs.getOps().get(0); - NBLabeledElement parent = NBLabeledElement.EMPTY; - return new ParsedOp(opTemplate, cfg, List.of(adapter.getPreprocessor()), parent); - } - - @Test - public void testQueryOpDispenserSimple() { - ParsedOp pop = parsedOpFor(""" - ops: - op1: - type: "query" - index: "test-index" - vector: "1.0,2.0,3.0" - namespace: "test-namespace" - top_k: 10 - filter: - filterfield: "field" - operator: "$gt" - comparator: 2.0 - include_values: true - include_metadata: true - query_vectors: - - id: 1 - values: "1.0,2.0,3.0" - top_k: 8 - namespace: "test-namespace" - filter: - filterfield: "field" - operator: "$lt" - comparator: 9.0 - sparse_values: - indices: "1,2,3" - values: "1.0,2.0,3.0" - - id: 2 - values: "4.0,5.0,6.0" - top_k: 11 - namespace: "test-namespace" - filter: - filterfield: "field" - operator: "$eq" - comparator: "val" - """); - OpDispenser dispenser = mapper.apply(pop); - assert(dispenser instanceof PineconeQueryOpDispenser); - PineconeOp op = dispenser.apply(0); - assert(op instanceof PineconeQueryOp); - } - - @Test - public void testDeleteOpDispenser() { - ParsedOp pop = parsedOpFor(""" - ops: - op1: - type: "delete" - index: "test-index" - ids: "1.0,2.0,3.0" - namespace: "test-namespace" - deleteall: true - filter: - filterfield: "key" - operator: "$eq" - comparator: "val" - """); - OpDispenser dispenser = mapper.apply(pop); - assert(dispenser instanceof PineconeDeleteOpDispenser); - PineconeOp op = dispenser.apply(0); - assert(op instanceof PineconeDeleteOp); - } - - @Test - public void testDescribeIndexStatsOpDispenser() { - ParsedOp pop = parsedOpFor(""" - ops: - op1: - type: "describeindexstats" - index: "test-index" - filter: - filterfield: "color" - operator: "$eq" - comparator: - - "green" - - "yellow" - - "red" - """); - OpDispenser dispenser = mapper.apply(pop); - assert(dispenser instanceof PineconeDescribeIndexStatsOpDispenser); - PineconeOp op = dispenser.apply(0); - assert(op instanceof PineconeDescribeIndexStatsOp); - - } - - @Test - public void testFetchOpDispenser() { - ParsedOp pop = parsedOpFor(""" - ops: - op1: - type: "fetch" - index: "test-index" - ids: "1.0,2.0,3.0" - namespace: "test-namespace" - """); - OpDispenser dispenser = mapper.apply(pop); - assert(dispenser instanceof PineconeFetchOpDispenser); - PineconeOp op = dispenser.apply(0); - assert(op instanceof PineconeFetchOp); - - } - - @Test - public void testUpdateOpDispenser() { - ParsedOp pop = parsedOpFor(""" - ops: - op1: - type: "update" - index: "test-index" - id: "id" - values: "1.0,2.0,3.0" - namespace: "test_namespace" - metadata: - key1: "val1" - key2: 2 - key3: 3 - sparse_values: - indices: "1,2,3" - values: "1.1,2.2,3.3" - """); - OpDispenser dispenser = mapper.apply(pop); - assert(dispenser instanceof PineconeUpdateOpDispenser); - PineconeOp op = dispenser.apply(0); - assert(op instanceof PineconeUpdateOp); - - } - - @Test - public void testUpsertOpDispenser() { - ParsedOp pop = parsedOpFor(""" - ops: - op1: - type: "upsert" - index: "test-index" - upsert_vectors: - - id: 1 - values: - - 1.0 - - 2.0 - - 3.0 - sparse_values: - indices: "1,2,3" - values: "4.0,5.0,6.0" - metadata: - key1: "val1" - key2: 2 - - id: 2 - values: - - 7.0 - - 8.0 - - 9.0 - sparse_values: - indices: "4,5,6" - values: "1.1,2.2,3.3" - """); - OpDispenser dispenser = mapper.apply(pop); - assert(dispenser instanceof PineconeUpsertOpDispenser); - PineconeOp op = dispenser.apply(0); - assert(op instanceof PineconeUpsertOp); - - } - -}