|
1 | 1 | package org.example; |
2 | 2 |
|
| 3 | +import static org.junit.jupiter.api.Assertions.assertEquals; |
| 4 | + |
3 | 5 | import org.json.JSONObject; |
4 | 6 | import org.junit.jupiter.api.Tag; |
5 | 7 | import org.junit.jupiter.api.Test; |
6 | 8 |
|
7 | | -import static org.junit.jupiter.api.Assertions.assertEquals; |
8 | | - |
9 | 9 | public class TokensTest { |
10 | 10 |
|
11 | | - @Test |
12 | | - @Tag("unit") |
13 | | - void testValidJSON() { |
14 | | - JSONObject meta = new JSONObject("{ `tokens`: { `input_tokens`: 69, `output_tokens`: 420 } }".replace('`', '"')); |
15 | | - Tokens tokens = Tokens.fromJson(meta); |
16 | | - assertEquals(tokens.getInputTokens(), 69); |
17 | | - assertEquals(tokens.getOutputTokens(), 420); |
18 | | - } |
| 11 | + @Test |
| 12 | + @Tag("unit") |
| 13 | + void testValidJSON() { |
| 14 | + JSONObject meta = |
| 15 | + new JSONObject( |
| 16 | + "{ `tokens`: { `input_tokens`: 69, `output_tokens`: 420 } }".replace('`', '"')); |
| 17 | + Tokens tokens = Tokens.fromJson(meta); |
| 18 | + assertEquals(tokens.getInputTokens(), 69); |
| 19 | + assertEquals(tokens.getOutputTokens(), 420); |
| 20 | + } |
19 | 21 |
|
20 | | - @Test |
21 | | - @Tag("unit") |
22 | | - void testJSONMissingTokens() { |
23 | | - JSONObject meta = new JSONObject("{}"); |
24 | | - Tokens tokens = Tokens.fromJson(meta); |
25 | | - assertEquals(tokens.getInputTokens(), 0); |
26 | | - assertEquals(tokens.getOutputTokens(), 0); |
27 | | - } |
| 22 | + @Test |
| 23 | + @Tag("unit") |
| 24 | + void testJSONMissingTokens() { |
| 25 | + JSONObject meta = new JSONObject("{}"); |
| 26 | + Tokens tokens = Tokens.fromJson(meta); |
| 27 | + assertEquals(tokens.getInputTokens(), 0); |
| 28 | + assertEquals(tokens.getOutputTokens(), 0); |
| 29 | + } |
28 | 30 |
|
29 | | - @Test |
30 | | - @Tag("unit") |
31 | | - void testJSONMissingInputTokens() { |
32 | | - JSONObject meta = new JSONObject("{ `tokens`: { `output_tokens`: 420 } }".replace('`', '"')); |
33 | | - Tokens tokens = Tokens.fromJson(meta); |
34 | | - assertEquals(tokens.getInputTokens(), 0); |
35 | | - assertEquals(tokens.getOutputTokens(), 420); |
36 | | - } |
| 31 | + @Test |
| 32 | + @Tag("unit") |
| 33 | + void testJSONMissingInputTokens() { |
| 34 | + JSONObject meta = new JSONObject("{ `tokens`: { `output_tokens`: 420 } }".replace('`', '"')); |
| 35 | + Tokens tokens = Tokens.fromJson(meta); |
| 36 | + assertEquals(tokens.getInputTokens(), 0); |
| 37 | + assertEquals(tokens.getOutputTokens(), 420); |
| 38 | + } |
37 | 39 |
|
38 | | - @Test |
39 | | - @Tag("unit") |
40 | | - void testJSONMissingOutputTokens() { |
41 | | - JSONObject meta = new JSONObject("{ `tokens`: { `input_tokens`: 69 } }".replace('`', '"')); |
42 | | - Tokens tokens = Tokens.fromJson(meta); |
43 | | - assertEquals(tokens.getInputTokens(), 69); |
44 | | - assertEquals(tokens.getOutputTokens(), 0); |
45 | | - } |
| 40 | + @Test |
| 41 | + @Tag("unit") |
| 42 | + void testJSONMissingOutputTokens() { |
| 43 | + JSONObject meta = new JSONObject("{ `tokens`: { `input_tokens`: 69 } }".replace('`', '"')); |
| 44 | + Tokens tokens = Tokens.fromJson(meta); |
| 45 | + assertEquals(tokens.getInputTokens(), 69); |
| 46 | + assertEquals(tokens.getOutputTokens(), 0); |
| 47 | + } |
46 | 48 | } |
0 commit comments