Skip to content

Commit

Permalink
*EmbeddingStoreTest -> *EmbeddingStoreIT
Browse files Browse the repository at this point in the history
  • Loading branch information
deep-learning-dynamo committed Nov 10, 2023
1 parent 4f7b574 commit f887190
Show file tree
Hide file tree
Showing 8 changed files with 10 additions and 12 deletions.
Original file line number Diff line number Diff line change
@@ -1,13 +1,11 @@
package dev.langchain4j.store.embedding.cassandra;

import com.datastax.astra.sdk.AstraClient;
import com.datastax.oss.driver.api.core.CqlSession;
import com.dtsx.astra.sdk.utils.TestUtils;
import dev.langchain4j.data.document.Metadata;
import dev.langchain4j.data.embedding.Embedding;
import dev.langchain4j.data.segment.TextSegment;
import dev.langchain4j.store.embedding.EmbeddingMatch;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.condition.EnabledIfEnvironmentVariable;

Expand All @@ -22,7 +20,7 @@
/**
* Testing implementation of Embedding Store using AstraDB.
*/
class AstraDbEmbeddingStoreTest {
class AstraDbEmbeddingStoreIT {

private static final String TEST_KEYSPACE = "langchain4j";
private static final String TEST_INDEX = "test_embedding_store";
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@
/**
* Work with Cassandra Embedding Store.
*/
class CassandraEmbeddingStoreTest {
class CassandraEmbeddingStoreIT {

@Test
@Disabled("To run this test, you must have a local Cassandra instance, a docker-compose is provided")
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@
import static org.assertj.core.data.Percentage.withPercentage;

@Disabled("needs Chroma running locally")
class ChromaEmbeddingStoreTest {
class ChromaEmbeddingStoreIT {

/**
* First ensure you have Chroma running locally. If not, then:
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@
import static org.assertj.core.data.Percentage.withPercentage;

@Disabled("needs Elasticsearch to be running locally")
class ElasticsearchEmbeddingStoreTest {
class ElasticsearchEmbeddingStoreIT {

/**
* First start elasticsearch locally:
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@
import static org.assertj.core.data.Percentage.withPercentage;

@Disabled("needs Milvus running locally")
class MilvusEmbeddingStoreTest {
class MilvusEmbeddingStoreIT {

/**
* First run Milvus locally:
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -18,12 +18,12 @@
import static org.assertj.core.data.Percentage.withPercentage;

@EnabledIfEnvironmentVariable(named = "PINECONE_API_KEY", matches = ".+")
class PineconeEmbeddingStoreTest {
class PineconeEmbeddingStoreIT {

private final PineconeEmbeddingStore embeddingStore = PineconeEmbeddingStore.builder()
.apiKey(System.getenv("PINECONE_API_KEY"))
.environment("asia-southeast1-gcp-free")
.projectId("75dc67a")
.environment("northamerica-northeast1-gcp")
.projectId("19a129b")
.index("test")
.nameSpace(randomUUID())
.build();
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@
import static org.assertj.core.data.Percentage.withPercentage;

@TestInstance(Lifecycle.PER_CLASS)
class RedisEmbeddingStoreTest {
class RedisEmbeddingStoreIT {

/**
* First start Redis locally:
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@
import static org.assertj.core.data.Percentage.withPercentage;

@EnabledIfEnvironmentVariable(named = "WEAVIATE_API_KEY", matches = ".+")
class WeaviateEmbeddingStoreTest {
class WeaviateEmbeddingStoreIT {

EmbeddingStore<TextSegment> embeddingStore = WeaviateEmbeddingStore.builder()
.apiKey(System.getenv("WEAVIATE_API_KEY"))
Expand Down

0 comments on commit f887190

Please sign in to comment.