tests: update tests to account for serialization and deserialization changes with new/modified model properties to support caching

This commit is contained in:
Stevan Freeborn
2024-08-16 09:21:43 -05:00
parent 66008717e1
commit 900a7a3549
10 changed files with 137 additions and 31 deletions
@@ -7,26 +7,39 @@ public class UsageTests : SerializationTest
{
var expectedInputTokens = 1;
var expectedOutputTokens = 2;
var expectedCacheCreationInputTokens = 3;
var expectedCacheReadInputTokens = 4;
var usage = new Usage
{
InputTokens = expectedInputTokens,
OutputTokens = expectedOutputTokens
OutputTokens = expectedOutputTokens,
CacheCreationInputTokens = expectedCacheCreationInputTokens,
CacheReadInputTokens = expectedCacheReadInputTokens
};
usage.InputTokens.Should().Be(expectedInputTokens);
usage.OutputTokens.Should().Be(expectedOutputTokens);
usage.CacheCreationInputTokens.Should().Be(expectedCacheCreationInputTokens);
usage.CacheReadInputTokens.Should().Be(expectedCacheReadInputTokens);
}
[Fact]
public void JsonSerialization_WhenCalled_ItShouldSerializeCorrectly()
{
var expectedJson = @"{ ""input_tokens"": 1, ""output_tokens"": 2 }";
var expectedJson = @"{
""input_tokens"": 1,
""output_tokens"": 2,
""cache_creation_input_tokens"": 3,
""cache_read_input_tokens"": 4
}";
var usage = new Usage
{
InputTokens = 1,
OutputTokens = 2
OutputTokens = 2,
CacheCreationInputTokens = 3,
CacheReadInputTokens = 4
};
var actual = Serialize(usage);
@@ -37,11 +50,18 @@ public class UsageTests : SerializationTest
[Fact]
public void JsonDeserialization_WhenCalled_ItShouldDeserializeCorrectly()
{
var json = @"{ ""input_tokens"": 1, ""output_tokens"": 2 }";
var json = @"{
""input_tokens"": 1,
""output_tokens"": 2,
""cache_creation_input_tokens"": 3,
""cache_read_input_tokens"": 4
}";
var usage = Deserialize<Usage>(json);
usage!.InputTokens.Should().Be(1);
usage.OutputTokens.Should().Be(2);
usage.CacheCreationInputTokens.Should().Be(3);
usage.CacheReadInputTokens.Should().Be(4);
}
}