TestStore

90 references to TestStore

Microsoft.Extensions.VectorData.ConformanceTests (90)

CollectionManagementTests.cs (5)

28await fixture.TestStore.DefaultVectorStore.EnsureCollectionDeletedAsync(collection.Name); 59await fixture.TestStore.DefaultVectorStore.EnsureCollectionDeletedAsync(collection.Name); 90public virtual string CollectionName => fixture.TestStore.AdjustCollectionName(CollectionNameBase); 100=> fixture.TestStore.CreateCollection<TKey, Record>(CollectionName, CreateRecordDefinition()); 110new VectorStoreVectorProperty(nameof(Record.Floats), typeof(ReadOnlyMemory<float>), 10) { IndexKind = fixture.TestStore.DefaultIndexKind }

DistanceFunctionTests.cs (4)

104await fixture.TestStore.WaitForDataAsync(collection, insertedRecords.Count, vectorSize: 4); 124if (!fixture.TestStore.SupportsScoreThreshold) 166public virtual string CollectionName => TestStore.AdjustCollectionName(CollectionNameBase); 188return TestStore.CreateCollection<TKey, SearchRecord>(CollectionName, definition);

EmbeddingGenerationTests.cs (5)

151var collection = stringVectorFixture.GetCollection<RawRecord>(stringVectorFixture.TestStore.DefaultVectorStore, stringVectorFixture.CollectionName + "withoutgenerator"); 201await stringVectorFixture.TestStore.WaitForDataAsync(collection, 1, filter: r => r.Counter == counter); 225await stringVectorFixture.TestStore.WaitForDataAsync(collection, 1, filter: r => (int)r[nameof(Record.Counter)] == counter); 258await stringVectorFixture.TestStore.WaitForDataAsync(collection, 2, filter: r => r.Counter == counter1 || r.Counter == counter2); 294await stringVectorFixture.TestStore.WaitForDataAsync(collection, 2, filter: r => (int)r[nameof(Record.Counter)] == counter1 || (int)r[nameof(Record.Counter)] == counter2);

FilterTests.cs (2)

580DynamicCollection = TestStore.CreateDynamicCollection(CollectionName, CreateRecordDefinition()); 723=> TestStore.WaitForDataAsync(Collection, recordCount: TestData.Count, r => r.Int > 0);

HybridSearchTests.cs (2)

206=> TestStore.WaitForDataAsync(Collection, recordCount: TestData.Count, vectorSize: 4); 256=> TestStore.WaitForDataAsync(Collection, recordCount: TestData.Count, vectorSize: 4);

IndexKindTests.cs (3)

47await fixture.TestStore.WaitForDataAsync(collection, records.Length); 58public virtual string CollectionName => TestStore.AdjustCollectionName(CollectionNameBase); 78return TestStore.CreateCollection<TKey, SearchRecord>(CollectionName, definition);

ModelTests\BasicModelTests.cs (16)

23expectedRecord.AssertEqual(received, includeVectors, fixture.TestStore.VectorsComparable); 40fixture.TestStore.VectorsComparable); 88fixture.TestStore.VectorsComparable); 111expectedRecord.AssertEqual(receivedRecord, includeVectors, fixture.TestStore.VectorsComparable); 223inserted.AssertEqual(received, includeVectors: true, fixture.TestStore.VectorsComparable); 225await fixture.TestStore.WaitForDataAsync(Collection, recordCount: fixture.TestData.Count + 1); 244updated.AssertEqual(received, includeVectors: true, fixture.TestStore.VectorsComparable); 277r => newRecords[0].AssertEqual(r, includeVectors: true, fixture.TestStore.VectorsComparable), 278r => newRecords[1].AssertEqual(r, includeVectors: true, fixture.TestStore.VectorsComparable)); 309r => existingRecords[0].AssertEqual(r, includeVectors: true, fixture.TestStore.VectorsComparable), 310r => existingRecords[1].AssertEqual(r, includeVectors: true, fixture.TestStore.VectorsComparable)); 341r => records[0].AssertEqual(r, includeVectors: true, fixture.TestStore.VectorsComparable), 342r => records[1].AssertEqual(r, includeVectors: true, fixture.TestStore.VectorsComparable)); 430expectedRecord.AssertEqual(result.Record, includeVectors, fixture.TestStore.VectorsComparable); 443fixture.TestData[1].AssertEqual(result.Record, includeVectors: false, fixture.TestStore.VectorsComparable); 456fixture.TestData[1].AssertEqual(result.Record, includeVectors: false, fixture.TestStore.VectorsComparable);

ModelTests\DynamicModelTests.cs (11)

27AssertEquivalent(expectedRecord, received, includeVectors, fixture.TestStore.VectorsComparable); 45fixture.TestStore.VectorsComparable); 96AssertEquivalent(expectedRecord, receivedRecord, includeVectors, fixture.TestStore.VectorsComparable); 226AssertEquivalent(inserted, received, includeVectors: true, fixture.TestStore.VectorsComparable); 245AssertEquivalent(updated, received, includeVectors: true, fixture.TestStore.VectorsComparable); 278r => AssertEquivalent(newRecords[0], r, includeVectors: true, fixture.TestStore.VectorsComparable), 279r => AssertEquivalent(newRecords[1], r, includeVectors: true, fixture.TestStore.VectorsComparable)); 323AssertEquivalent(expectedRecord, result.Record, includeVectors, fixture.TestStore.VectorsComparable); 336AssertEquivalent(fixture.TestData[1], result.Record, includeVectors: false, fixture.TestStore.VectorsComparable); 349AssertEquivalent(fixture.TestData[1], result.Record, includeVectors: false, fixture.TestStore.VectorsComparable); 395=> TestStore.CreateDynamicCollection(CollectionName, CreateRecordDefinition());

ModelTests\MultiVectorModelTests.cs (5)

24expectedRecord.AssertEqual(received, includeVectors, fixture.TestStore.VectorsComparable); 43inserted.AssertEqual(received, includeVectors: true, fixture.TestStore.VectorsComparable); 61fixture.TestData[0].AssertEqual(result.Record, includeVectors: true, fixture.TestStore.VectorsComparable); 66fixture.TestData[1].AssertEqual(result.Record, includeVectors: true, fixture.TestStore.VectorsComparable); 125=> TestStore.WaitForDataAsync(Collection, recordCount: TestData.Count, vectorProperty: r => r.Vector1);

ModelTests\NoDataModelTests.cs (2)

24expectedRecord.AssertEqual(received, includeVectors, fixture.TestStore.VectorsComparable); 41inserted.AssertEqual(received, includeVectors: true, fixture.TestStore.VectorsComparable);

Support\VectorStoreCollectionFixtureBase.cs (5)

30public virtual string CollectionName => TestStore.AdjustCollectionName(CollectionNameBase); 32protected virtual string DistanceFunction => TestStore.DefaultDistanceFunction; 33protected virtual string IndexKind => TestStore.DefaultIndexKind; 36=> TestStore.CreateCollection<TKey, TRecord>(CollectionName, CreateRecordDefinition()); 67=> TestStore.WaitForDataAsync(Collection, recordCount: TestData.Count);

Support\VectorStoreFixture.cs (8)

14public virtual VectorStore VectorStore => TestStore.DefaultVectorStore; 16public virtual string DefaultDistanceFunction => TestStore.DefaultDistanceFunction; 17public virtual string DefaultIndexKind => TestStore.DefaultIndexKind; 20=> new(TestStore.ReferenceCountingStartAsync()); 25return new(TestStore.ReferenceCountingStopAsync()); 29=> TestStore.GenerateKey<TKey>(Interlocked.Increment(ref _nextKeyValue)); 40=> TestStore.CreateCollection<TKey, TRecord>(name, definition); 49=> TestStore.CreateDynamicCollection(name, definition);

TypeTests\DataTypeTests.cs (6)

172await fixture.TestStore.WaitForDataAsync(fixture.Collection, recordCount: 0); 197await fixture.TestStore.WaitForDataAsync(fixture.Collection, recordCount: 0); 265await fixture.TestStore.WaitForDataAsync(fixture.Collection, recordCount: testData.Count); 312await fixture.TestStore.WaitForDataAsync(dynamicCollection, recordCount: testData.Count); 360await fixture.TestStore.WaitForDataAsync(collection, filter: filter, recordCount: 1); 401await fixture.TestStore.WaitForDataAsync(dynamicCollection, filter: filter, recordCount: 1);

TypeTests\EmbeddingTypeTests.cs (8)

66await fixture.TestStore.WaitForDataAsync(collection, recordCount: 1, filter: r => r.Int == 42, dummyVector: value); 70if (fixture.TestStore.VectorsComparable) 79if (fixture.TestStore.VectorsComparable) 113await fixture.TestStore.WaitForDataAsync(dynamicCollection, recordCount: 1, filter: r => (int)r["Int"]! == 43, dummyVector: value); 117if (fixture.TestStore.VectorsComparable) 126if (fixture.TestStore.VectorsComparable) 162await fixture.TestStore.WaitForDataAsync(collection2, recordCount: 1, filter: r => r.Int == 44, dummyVector: value); 226public virtual string CollectionName => TestStore.AdjustCollectionName(CollectionNameBase);

TypeTests\KeyTypeTests.cs (8)

30fixture.TestStore.CreateCollection<Guid, Record<string>>( 75await fixture.TestStore.WaitForDataAsync(collection, recordCount: 2); 99await fixture.TestStore.WaitForDataAsync(collection, recordCount: 0); 122await fixture.TestStore.WaitForDataAsync(dynamicCollection, recordCount: 2); 194await fixture.TestStore.WaitForDataAsync(collectionWithAutoGeneration, recordCount: 4); 221public virtual string CollectionName => TestStore.AdjustCollectionName(CollectionNameBase); 225=> TestStore.CreateCollection<TKey, Record<TKey>>(CollectionName, CreateRecordDefinition<TKey>(withAutoGeneration)); 229=> TestStore.CreateDynamicCollection(CollectionName, CreateRecordDefinition<TKey>(withAutoGeneration));