From d50af2e474281a1e88d191f783c2792824698974 Mon Sep 17 00:00:00 2001 From: hachem Date: Thu, 10 Sep 2026 05:52:08 +0200 Subject: feat: conditional select --- tests/codegen_test.c | 37 +++++++++++++++++++++++++++++++++++++ 1 file changed, 37 insertions(+) (limited to 'tests/codegen_test.c') diff --git a/tests/codegen_test.c b/tests/codegen_test.c index 2b5fdb7..0a4a841 100644 --- a/tests/codegen_test.c +++ b/tests/codegen_test.c @@ -104,6 +104,41 @@ static void test_generate_fasm(struct TestContext* context) free_program(&program); } +static void test_generate_select(struct TestContext* context) +{ + // destination distinct from both sources: default-move else, then cmov + struct Lexer lexer = create_lexer("proc main\n{\nrax = rbx if rcx < rdx else rsi\n}\n"); + struct Program program; + check(context, parse_program(&lexer, &program)); + + char buffer[1024]; + generate_to_buffer(&program, buffer, sizeof(buffer)); + + check(context, strstr(buffer, "mov rax, rsi") != NULL); + check(context, strstr(buffer, "cmp rcx, rdx") != NULL); + check(context, strstr(buffer, "cmovl rax, rbx") != NULL); + check(context, strstr(buffer, "; TODO") == NULL); + + free_program(&program); +} + +static void test_generate_select_aarch64(struct TestContext* context) +{ + struct Lexer lexer = create_lexer( + "[enable: logical_registers]\nproc main\n{\nr1 = r2 if r3 < r4 else r5\n}\n"); + struct Program program; + check(context, parse_program(&lexer, &program)); + + char buffer[1024]; + generate_aarch64_to_buffer(&program, buffer, sizeof(buffer)); + + check(context, strstr(buffer, "cmp x2, x3") != NULL); + check(context, strstr(buffer, "csel x0, x1, x4, lt") != NULL); + check(context, strstr(buffer, "; TODO") == NULL); + + free_program(&program); +} + static void test_generate_aarch64(struct TestContext* context) { // logical registers map r1 -> x0, r2 -> x1; three-operand arithmetic and svc @@ -633,6 +668,8 @@ void run_codegen_tests(struct TestContext* context) test_generate_consts_and_data(context); test_generate_fasm(context); test_generate_aarch64(context); + test_generate_select(context); + test_generate_select_aarch64(context); test_generate_text(context); test_generate_if(context); test_generate_if_else(context); -- cgit v1.3