java-llama.cpp/src/test/java/net/ladenthin/llama/args/ModelFlagTest.java at b049986dfab80033c6839ebb9ca93e1b705176aa · bernardladenthin/java-llama.cpp · GitHub

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
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
// SPDX-FileCopyrightText: 2026 Bernard Ladenthin <bernard.ladenthin@gmail.com>
// SPDX-FileCopyrightText: 2023-2025 Konstantin Herud
//
// SPDX-License-Identifier: MIT

package net.ladenthin.llama.args;

import static org.junit.jupiter.api.Assertions.*;

import java.util.Arrays;
import java.util.Collection;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.params.ParameterizedTest;
import org.junit.jupiter.params.provider.MethodSource;

public class ModelFlagTest {

    public static Collection<Object[]> data() {
        return Arrays.asList(new Object[][] {
            {ModelFlag.NO_CONTEXT_SHIFT, "--no-context-shift"},
            {ModelFlag.FLASH_ATTN, "--flash-attn"},
            {ModelFlag.SWA_FULL, "--swa-full"},
            {ModelFlag.NO_PERF, "--no-perf"},
            {ModelFlag.ESCAPE, "--escape"},
            {ModelFlag.NO_ESCAPE, "--no-escape"},
            {ModelFlag.SPECIAL, "--special"},
            {ModelFlag.NO_WARMUP, "--no-warmup"},
            {ModelFlag.SPM_INFILL, "--spm-infill"},
            {ModelFlag.IGNORE_EOS, "--ignore-eos"},
            {ModelFlag.DUMP_KV_CACHE, "--dump-kv-cache"},
            {ModelFlag.NO_KV_OFFLOAD, "--no-kv-offload"},
            {ModelFlag.CONT_BATCHING, "--cont-batching"},
            {ModelFlag.NO_CONT_BATCHING, "--no-cont-batching"},
            {ModelFlag.MLOCK, "--mlock"},
            {ModelFlag.NO_MMAP, "--no-mmap"},
            {ModelFlag.CHECK_TENSORS, "--check-tensors"},
            {ModelFlag.EMBEDDING, "--embedding"},
            {ModelFlag.RERANKING, "--reranking"},
            {ModelFlag.LORA_INIT_WITHOUT_APPLY, "--lora-init-without-apply"},
            {ModelFlag.LOG_DISABLE, "--log-disable"},
            {ModelFlag.VERBOSE, "--verbose"},
            {ModelFlag.LOG_PREFIX, "--log-prefix"},
            {ModelFlag.LOG_TIMESTAMPS, "--log-timestamps"},
            {ModelFlag.JINJA, "--jinja"},
            {ModelFlag.VOCAB_ONLY, "--vocab-only"},
            {ModelFlag.KV_UNIFIED, "--kv-unified"},
            {ModelFlag.NO_KV_UNIFIED, "--no-kv-unified"},
            {ModelFlag.CLEAR_IDLE, "--cache-idle-slots"},
            {ModelFlag.NO_CLEAR_IDLE, "--no-cache-idle-slots"},
            {ModelFlag.MMPROJ_AUTO, "--mmproj-auto"},
            {ModelFlag.NO_MMPROJ_AUTO, "--no-mmproj-auto"},
            {ModelFlag.MMPROJ_OFFLOAD, "--mmproj-offload"},
            {ModelFlag.NO_MMPROJ_OFFLOAD, "--no-mmproj-offload"},
            {ModelFlag.OFFLINE, "--offline"},
        });
    }

    @ParameterizedTest(name = "{0} -> {1}")
    @MethodSource("data")
    public void testGetCliFlag(ModelFlag flag, String expectedCliFlag) {
        assertEquals(expectedCliFlag, flag.getCliFlag());
    }

    // ------------------------------------------------------------------
    // Structural invariants
    // ------------------------------------------------------------------

    @Test
    public void testEnumCount() {
        assertEquals(35, ModelFlag.values().length);
    }

    @ParameterizedTest(name = "{0} -> {1}")
    @MethodSource("data")
    public void testCliFlagStartsWithDoubleDash(ModelFlag flag, String expectedCliFlag) {
        assertTrue(flag.getCliFlag().startsWith("--"), "Flag " + flag + " must start with --");
    }

    @ParameterizedTest(name = "{0} -> {1}")
    @MethodSource("data")
    public void testCliFlagNonEmpty(ModelFlag flag, String expectedCliFlag) {
        assertFalse(flag.getCliFlag().isEmpty(), "Flag " + flag + " has empty CLI string");
    }
}