diff --git a/LLama.Unittest/LLamaEmbedderTests.cs b/LLama.Unittest/LLamaEmbedderTests.cs index cea946d1..ec132dd0 100644 --- a/LLama.Unittest/LLamaEmbedderTests.cs +++ b/LLama.Unittest/LLamaEmbedderTests.cs @@ -18,14 +18,24 @@ public class LLamaEmbedderTests return a.Zip(b, (x, y) => x + y).Sum(); } + private static void AssertApproxStartsWith(float[] array, float[] start, float epsilon = 0.00001f) + { + for (int i = 0; i < start.Length; i++) + Assert.Equal(array[i], start[i], epsilon); + } + [Fact] - public void EmbedHello() + public void EmbedBasic() { - var hello = _embedder.GetEmbeddings("Hello"); + var hello = _embedder.GetEmbeddings("cat"); Assert.NotNull(hello); Assert.NotEmpty(hello); //Assert.Equal(_embedder.EmbeddingSize, hello.Length); + + // Expected value generate with llama.cpp embedding.exe + var expected = new float[] { -0.127304f, -0.678057f, -0.085244f, -0.956915f, -0.638633f }; + AssertApproxStartsWith(hello, expected); } [Fact]