wehub-resource-sync
97 lines
3.4 KiB
C#
97 lines
3.4 KiB
C#
// Copyright (c) Microsoft. All rights reserved.
|
|
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using Microsoft.Extensions.VectorData;
|
|
using Microsoft.Extensions.VectorData.ProviderServices;
|
|
using Microsoft.SemanticKernel.Connectors.SqliteVec;
|
|
using Xunit;
|
|
|
|
namespace SemanticKernel.Connectors.SqliteVec.UnitTests;
|
|
|
|
/// <summary>
|
|
/// Unit tests for <see cref="SqlitePropertyMapping"/> class.
|
|
/// </summary>
|
|
public sealed class SqlitePropertyMappingTests
|
|
{
|
|
[Fact]
|
|
public void MapVectorForStorageModelReturnsByteArray()
|
|
{
|
|
// Arrange
|
|
var vector = new ReadOnlyMemory<float>([1.1f, 2.2f, 3.3f, 4.4f]);
|
|
|
|
// Act
|
|
var storageModelVector = SqlitePropertyMapping.MapVectorForStorageModel(vector);
|
|
|
|
// Assert
|
|
Assert.IsType<byte[]>(storageModelVector);
|
|
Assert.True(storageModelVector.Length > 0);
|
|
}
|
|
|
|
[Theory]
|
|
[InlineData(true)]
|
|
[InlineData(false)]
|
|
public void GetColumnsReturnsCollectionOfColumns(bool data)
|
|
{
|
|
// Arrange
|
|
var properties = new List<PropertyModel>()
|
|
{
|
|
new KeyPropertyModel("Key", typeof(string)) { StorageName = "Key" },
|
|
new DataPropertyModel("Data", typeof(int)) { StorageName = "my_data", IsIndexed = true },
|
|
new DataPropertyModel("NullableData", typeof(int?)) { StorageName = "nullable_data" },
|
|
new DataPropertyModel("StringData", typeof(string)) { StorageName = "string_data" },
|
|
new VectorPropertyModel("Vector", typeof(ReadOnlyMemory<float>))
|
|
{
|
|
Dimensions = 4,
|
|
DistanceFunction = DistanceFunction.ManhattanDistance,
|
|
StorageName = "Vector"
|
|
}
|
|
};
|
|
|
|
// Act
|
|
var columns = SqlitePropertyMapping.GetColumns(properties, data: data);
|
|
|
|
// Assert
|
|
Assert.Equal("Key", columns[0].Name);
|
|
Assert.Equal("TEXT", columns[0].Type);
|
|
Assert.True(columns[0].IsPrimary);
|
|
Assert.Null(columns[0].Configuration);
|
|
Assert.False(columns[0].HasIndex);
|
|
// string key is a reference type, so nullable
|
|
Assert.True(columns[0].IsNullable);
|
|
|
|
if (data)
|
|
{
|
|
Assert.Equal("my_data", columns[1].Name);
|
|
Assert.Equal("INTEGER", columns[1].Type);
|
|
Assert.False(columns[1].IsPrimary);
|
|
Assert.Null(columns[1].Configuration);
|
|
Assert.True(columns[1].HasIndex);
|
|
// int is a non-nullable value type
|
|
Assert.False(columns[1].IsNullable);
|
|
|
|
Assert.Equal("nullable_data", columns[2].Name);
|
|
Assert.Equal("INTEGER", columns[2].Type);
|
|
Assert.False(columns[2].IsPrimary);
|
|
// int? is a nullable value type
|
|
Assert.True(columns[2].IsNullable);
|
|
|
|
Assert.Equal("string_data", columns[3].Name);
|
|
Assert.Equal("TEXT", columns[3].Type);
|
|
// string is a reference type, so nullable
|
|
Assert.True(columns[3].IsNullable);
|
|
}
|
|
else
|
|
{
|
|
Assert.Equal("Vector", columns[1].Name);
|
|
Assert.Equal("FLOAT[4]", columns[1].Type);
|
|
Assert.False(columns[1].IsPrimary);
|
|
Assert.NotNull(columns[1].Configuration);
|
|
Assert.False(columns[1].HasIndex);
|
|
Assert.Equal("l1", columns[1].Configuration!["distance_metric"]);
|
|
// ReadOnlyMemory<float> is a non-nullable value type
|
|
Assert.False(columns[1].IsNullable);
|
|
}
|
|
}
|
|
}
|