Browse Source

Added a known correct value to the unit tests

tags/v0.5.1
Martin Evans 2 years ago
parent
commit
35e48691ea
1 changed files with 12 additions and 2 deletions
  1. +12
    -2
      LLama.Unittest/LLamaEmbedderTests.cs

+ 12
- 2
LLama.Unittest/LLamaEmbedderTests.cs View File

@@ -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]


Loading…
Cancel
Save