tokrs 0.1.0

A CLI tool for counting tokens
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39

use super::*;

#[test]
fn test_thousands() {
    assert_eq!(thousands(0), "0");
    assert_eq!(thousands(999), "999");
    assert_eq!(thousands(1_000), "1,000");
    assert_eq!(thousands(76_780_408), "76,780,408");
}

#[test]
fn test_cost_text() {
    let all_unpriced = TokenTotals {
        requests: 3,
        unpriced: 3,
        ..Default::default()
    };
    assert_eq!(cost_text(&all_unpriced), "-");
    let partial = TokenTotals {
        requests: 3,
        cost_usd: 1.5,
        unpriced: 1,
        ..Default::default()
    };
    assert_eq!(cost_text(&partial), "$1.50*");
    let small = TokenTotals {
        requests: 1,
        cost_usd: 0.0042,
        ..Default::default()
    };
    assert_eq!(cost_text(&small), "$0.0042");
    // 已定价部分合计恰为 0 且仍有 unpriced: 显示 $0.00* 而非 "-"
    let zero_priced = TokenTotals {
        requests: 2,
        unpriced: 1,
        ..Default::default()
    };
    assert_eq!(cost_text(&zero_priced), "$0.00*");
}