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
@@ -10,7 +10,9 @@ public class MessageDeltaEventDataTests : SerializationTest
},
""usage"": {
""input_tokens"": 1,
""output_tokens"": 1
""output_tokens"": 1,
""cache_creation_input_tokens"": 1,
""cache_read_input_tokens"": 1
}
}";
@@ -18,7 +20,13 @@ public class MessageDeltaEventDataTests : SerializationTest
public void Constructor_WhenCalled_ItShouldInitializeProperties()
{
var expectedDelta = new MessageDelta("max_tokens", "max_tokens");
var expectedUsage = new Usage { InputTokens = 1, OutputTokens = 1 };
var expectedUsage = new Usage
{
InputTokens = 1,
OutputTokens = 1,
CacheCreationInputTokens = 1,
CacheReadInputTokens = 1,
};
var messageDeltaEventData = new MessageDeltaEventData(expectedDelta, expectedUsage);
@@ -30,7 +38,13 @@ public class MessageDeltaEventDataTests : SerializationTest
public void JsonSerialization_WhenSerialized_ItShouldHaveExpectedShape()
{
var expectedDelta = new MessageDelta("max_tokens", "max_tokens");
var expectedUsage = new Usage { InputTokens = 1, OutputTokens = 1 };
var expectedUsage = new Usage
{
InputTokens = 1,
OutputTokens = 1,
CacheCreationInputTokens = 1,
CacheReadInputTokens = 1,
};
var messageDeltaEventData = new MessageDeltaEventData(expectedDelta, expectedUsage);
@@ -43,7 +57,13 @@ public class MessageDeltaEventDataTests : SerializationTest
public void JsonDeserialization_WhenDeserialized_ItShouldHaveExpectedValues()
{
var expectedDelta = new MessageDelta("max_tokens", "max_tokens");
var expectedUsage = new Usage { InputTokens = 1, OutputTokens = 1 };
var expectedUsage = new Usage
{
InputTokens = 1,
OutputTokens = 1,
CacheCreationInputTokens = 1,
CacheReadInputTokens = 1,
};
var messageDeltaEventData = Deserialize<MessageDeltaEventData>(_testJson);