GitVita
New repositoryImport repositoryNew issue
Your repositories Your dashboard Sign in

openai/openai-dotnet

Public

mirrored from https://github.com/openai/openai-dotnetAvailable

Watch0Fork0Star0
CodeCommitsIssuesPull requestsActionsInsightsSecurity
OpenAI_2.0.0-beta.10

Branches

Tags

  • No tags available.
0Branches0Tags
Go to file
Add file
Code

Clone

HTTPS

Download ZIP

openai-dotnet/tests/Embeddings

tests/Embeddings/OpenAIEmbeddingsModelFactoryTests.cs

115lines · modecode

RawDownload
Latest commit unavailable.
1using System.Collections.Generic;
2using System.Linq;
3using NUnit.Framework;
4using OpenAI.Embeddings;
5
6namespace OpenAI.Tests.Embeddings;
7
8[Parallelizable(ParallelScope.All)]
9[Category("Smoke")]
10public partial class OpenAIEmbeddingsModelFactoryTests
11{
12 [Test]
13 public void EmbeddingWithNoPropertiesWorks()
14 {
15 Embedding embedding = OpenAIEmbeddingsModelFactory.Embedding();
16
17 Assert.That(embedding.Index, Is.EqualTo(default(int)));
18 Assert.That(embedding.Vector.ToArray(), Is.Not.Null.And.Empty);
19 }
20
21 [Test]
22 public void EmbeddingWithIndexWorks()
23 {
24 int index = 10;
25 Embedding embedding = OpenAIEmbeddingsModelFactory.Embedding(index: index);
26
27 Assert.That(embedding.Index, Is.EqualTo(index));
28 Assert.That(embedding.Vector.ToArray(), Is.Not.Null.And.Empty);
29 }
30
31 [Test]
32 public void EmbeddingWithVectorWorks()
33 {
34 IEnumerable<float> vector = [ 1f, 2f, 3f ];
35 Embedding embedding = OpenAIEmbeddingsModelFactory.Embedding(vector: vector);
36
37 Assert.That(embedding.Index, Is.EqualTo(default(int)));
38 Assert.That(embedding.Vector.ToArray().SequenceEqual(vector), Is.True);
39 }
40
41 [Test]
42 public void EmbeddingCollectionWithNoPropertiesWorks()
43 {
44 EmbeddingCollection embeddingCollection = OpenAIEmbeddingsModelFactory.EmbeddingCollection();
45
46 Assert.That(embeddingCollection.Count, Is.EqualTo(0));
47 Assert.That(embeddingCollection.Model, Is.Null);
48 Assert.That(embeddingCollection.Usage, Is.Null);
49 }
50
51 [Test]
52 public void EmbeddingCollectionWithItemsWorks()
53 {
54 IEnumerable<Embedding> items = [
55 OpenAIEmbeddingsModelFactory.Embedding(index: 10),
56 OpenAIEmbeddingsModelFactory.Embedding(index: 20)
57 ];
58 EmbeddingCollection embeddingCollection = OpenAIEmbeddingsModelFactory.EmbeddingCollection(items: items);
59
60 Assert.That(embeddingCollection.SequenceEqual(items), Is.True);
61 Assert.That(embeddingCollection.Model, Is.Null);
62 Assert.That(embeddingCollection.Usage, Is.Null);
63 }
64
65 [Test]
66 public void EmbeddingCollectionWithModelWorks()
67 {
68 string model = "supermodel";
69 EmbeddingCollection embeddingCollection = OpenAIEmbeddingsModelFactory.EmbeddingCollection(model: model);
70
71 Assert.That(embeddingCollection.Count, Is.EqualTo(0));
72 Assert.That(embeddingCollection.Model, Is.EqualTo(model));
73 Assert.That(embeddingCollection.Usage, Is.Null);
74 }
75
76 [Test]
77 public void EmbeddingCollectionWithUsageWorks()
78 {
79 EmbeddingTokenUsage usage = OpenAIEmbeddingsModelFactory.EmbeddingTokenUsage(inputTokens: 10);
80 EmbeddingCollection embeddingCollection = OpenAIEmbeddingsModelFactory.EmbeddingCollection(usage: usage);
81
82 Assert.That(embeddingCollection.Count, Is.EqualTo(0));
83 Assert.That(embeddingCollection.Model, Is.Null);
84 Assert.That(embeddingCollection.Usage, Is.EqualTo(usage));
85 }
86
87 [Test]
88 public void EmbeddingTokenUsageWithNoPropertiesWorks()
89 {
90 EmbeddingTokenUsage embeddingTokenUsage = OpenAIEmbeddingsModelFactory.EmbeddingTokenUsage();
91
92 Assert.That(embeddingTokenUsage.InputTokens, Is.EqualTo(default(int)));
93 Assert.That(embeddingTokenUsage.TotalTokens, Is.EqualTo(default(int)));
94 }
95
96 [Test]
97 public void EmbeddingTokenUsageWithInputTokensWorks()
98 {
99 int inputTokens = 10;
100 EmbeddingTokenUsage embeddingTokenUsage = OpenAIEmbeddingsModelFactory.EmbeddingTokenUsage(inputTokens: inputTokens);
101
102 Assert.That(embeddingTokenUsage.InputTokens, Is.EqualTo(10));
103 Assert.That(embeddingTokenUsage.TotalTokens, Is.EqualTo(default(int)));
104 }
105
106 [Test]
107 public void EmbeddingTokenUsageWithTotalTokensWorks()
108 {
109 int totalTokens = 10;
110 EmbeddingTokenUsage embeddingTokenUsage = OpenAIEmbeddingsModelFactory.EmbeddingTokenUsage(totalTokens: totalTokens);
111
112 Assert.That(embeddingTokenUsage.InputTokens, Is.EqualTo(default(int)));
113 Assert.That(embeddingTokenUsage.TotalTokens, Is.EqualTo(totalTokens));
114 }
115}
116