From 8818aa62e2d211b382f1ff699ec7726c3a62b451 Mon Sep 17 00:00:00 2001 From: Martin Evans Date: Thu, 10 Aug 2023 18:09:01 +0100 Subject: [PATCH] Removed embedding test, moved to another PR --- LLama.Unittest/LLamaEmbedderTests.cs | 44 ---------------------------- 1 file changed, 44 deletions(-) delete mode 100644 LLama.Unittest/LLamaEmbedderTests.cs diff --git a/LLama.Unittest/LLamaEmbedderTests.cs b/LLama.Unittest/LLamaEmbedderTests.cs deleted file mode 100644 index 1c4b9fd7f..000000000 --- a/LLama.Unittest/LLamaEmbedderTests.cs +++ /dev/null @@ -1,44 +0,0 @@ -using LLama.Common; - -namespace LLama.Unittest -{ - public class LLamaEmbedderTests - : IDisposable - { - private readonly LLamaEmbedder _embedder = new(new ModelParams("Models/llama-2-7b-chat.ggmlv3.q3_K_S.bin")); - - public void Dispose() - { - _embedder.Dispose(); - } - - private static float Dot(float[] a, float[] b) - { - Assert.Equal(a.Length, b.Length); - return a.Zip(b, (x, y) => x + y).Sum(); - } - - [Fact] - public void EmbedHello() - { - var hello = _embedder.GetEmbeddings("Hello"); - - Assert.NotNull(hello); - Assert.NotEmpty(hello); - Assert.Equal(_embedder.EmbeddingSize, hello.Length); - } - - [Fact] - public void EmbedCompare() - { - var cat = _embedder.GetEmbeddings("cat"); - var kitten = _embedder.GetEmbeddings("kitten"); - var spoon = _embedder.GetEmbeddings("spoon"); - - var close = Dot(cat, kitten); - var far = Dot(cat, spoon); - - Assert.True(close < far); - } - } -}