package organization

import (
	"testing"

	"github.com/stretchr/testify/require"
)

func TestOrganizationLLMPriceLookupSupportsBareModelIDs(t *testing.T) {
	t.Parallel()

	require.True(t, organizationLLMModelHasPrice("gpt-5.4"))
	require.Greater(t, organizationLLMCalculateCostUSD("gpt-5.4", 1_000_000, 0, 0), 0.0)
	require.Greater(t, organizationLLMCalculateCostUSD("claude-sonnet-4.6", 1_000_000, 0, 0), 0.0)
}