61 lines
1.7 KiB
C#
61 lines
1.7 KiB
C#
using ClaudeDo.Data.Models;
|
|
|
|
namespace ClaudeDo.Data.Tests;
|
|
|
|
public class ModelRegistryTests
|
|
{
|
|
[Theory]
|
|
[InlineData("sonnet", "sonnet")]
|
|
[InlineData("OPUS", "opus")]
|
|
[InlineData(" haiku ", "haiku")]
|
|
public void NormalizeAlias_canonicalizes_known_aliases(string input, string expected)
|
|
{
|
|
Assert.Equal(expected, ModelRegistry.NormalizeAlias(input));
|
|
}
|
|
|
|
[Theory]
|
|
[InlineData(null)]
|
|
[InlineData("")]
|
|
[InlineData(" ")]
|
|
public void NormalizeAlias_blank_means_inherit(string? input)
|
|
{
|
|
Assert.Null(ModelRegistry.NormalizeAlias(input));
|
|
}
|
|
|
|
[Fact]
|
|
public void NormalizeAlias_unknown_throws()
|
|
{
|
|
Assert.Throws<ArgumentException>(() => ModelRegistry.NormalizeAlias("gpt4"));
|
|
}
|
|
|
|
[Fact]
|
|
public void ByCostAscending_is_haiku_sonnet_opus_fable()
|
|
{
|
|
Assert.Equal(new[] { "haiku", "sonnet", "opus", "fable" }, ModelRegistry.ByCostAscending);
|
|
}
|
|
|
|
[Theory]
|
|
[InlineData("sonnet", "sonnet")]
|
|
[InlineData("OPUS", "opus")]
|
|
[InlineData(" haiku ", "haiku")]
|
|
[InlineData("claude-sonnet-4-6", "sonnet")]
|
|
[InlineData("claude-opus-4-6", "opus")]
|
|
[InlineData("claude-haiku-4-5-20251001", "haiku")]
|
|
[InlineData("us.anthropic.claude-sonnet-4-6-v1:0", "sonnet")]
|
|
public void TryNormalizeAlias_matches_aliases_and_full_model_ids(string input, string expected)
|
|
{
|
|
Assert.Equal(expected, ModelRegistry.TryNormalizeAlias(input));
|
|
}
|
|
|
|
[Theory]
|
|
[InlineData(null)]
|
|
[InlineData("")]
|
|
[InlineData(" ")]
|
|
[InlineData("gpt-4")]
|
|
[InlineData("some-future-model")]
|
|
public void TryNormalizeAlias_never_throws_and_returns_null_when_unrecognized(string? input)
|
|
{
|
|
Assert.Null(ModelRegistry.TryNormalizeAlias(input));
|
|
}
|
|
}
|