4 files changed, 120 insertions, 1 deletions

diff --git a/tests/codegen_test.c b/tests/codegen_test.c
index 61e441f..377ddeb 100644
--- a/tests/codegen_test.c
+++ b/tests/codegen_test.c
@@ -426,8 +426,109 @@ static void test_generate_enum_struct(struct TestContext* context)
 	free_program(&program);
 }
 
+static void test_generate_floats(struct TestContext* context)
+{
+	struct Lexer lexer = create_lexer(
+		"proc main\n{\nxmm0 = 3.5\nxmm0 *= xmm1\nrax = 4\nxmm2 = rax\nrbx = xmm0\n}\n");
+	struct Program program;
+	check(context, parse_program(&lexer, &program));
+
+	FILE* out = tmpfile();
+	generate_nasm(&program, out);
+	fflush(out);
+	rewind(out);
+
+	char buffer[1024];
+	size_t read = fread(buffer, 1, sizeof(buffer) - 1, out);
+	buffer[read] = '\0';
+	fclose(out);
+
+	check(context, strstr(buffer, "__float0: dq 3.5") != NULL);
+	check(context, strstr(buffer, "movsd xmm0, [__float0]") != NULL);
+	check(context, strstr(buffer, "mulsd xmm0, xmm1") != NULL);   // float arithmetic
+	check(context, strstr(buffer, "cvtsi2sd xmm2, rax") != NULL); // int -> float
+	check(context, strstr(buffer, "cvttsd2si rbx, xmm0") != NULL); // float -> int
+	check(context, strstr(buffer, "; TODO") == NULL);
+
+	free_program(&program);
+}
+
+static void test_generate_float_compare(struct TestContext* context)
+{
+	struct Lexer lexer = create_lexer("proc main\n{\nif xmm0 > 4.0\ngoto done\ndone:\nsyscall\n}\n");
+	struct Program program;
+	check(context, parse_program(&lexer, &program));
+
+	FILE* out = tmpfile();
+	generate_nasm(&program, out);
+	fflush(out);
+	rewind(out);
+
+	char buffer[1024];
+	size_t read = fread(buffer, 1, sizeof(buffer) - 1, out);
+	buffer[read] = '\0';
+	fclose(out);
+
+	check(context, strstr(buffer, "ucomisd xmm0, [__float0]") != NULL);
+	check(context, strstr(buffer, "jbe .if_end") != NULL); // '>' skips when <=
+	check(context, strstr(buffer, "; TODO") == NULL);
+
+	free_program(&program);
+}
+
+static void test_generate_add_zero_peephole(struct TestContext* context)
+{
+	// `+= 0` / `-= 0` (e.g. a struct field at offset 0) is dropped; `*= 0` is not
+	struct Lexer lexer = create_lexer("proc main\n{\nrax += 0\nrbx -= 0\nrcx *= 0\n}\n");
+	struct Program program;
+	check(context, parse_program(&lexer, &program));
+
+	FILE* out = tmpfile();
+	generate_nasm(&program, out);
+	fflush(out);
+	rewind(out);
+
+	char buffer[1024];
+	size_t read = fread(buffer, 1, sizeof(buffer) - 1, out);
+	buffer[read] = '\0';
+	fclose(out);
+
+	check(context, strstr(buffer, "add rax, 0") == NULL);
+	check(context, strstr(buffer, "sub rbx, 0") == NULL);
+	check(context, strstr(buffer, "imul rcx, 0") != NULL);
+
+	free_program(&program);
+}
+
+static void test_generate_float_memory(struct TestContext* context)
+{
+	struct Lexer lexer = create_lexer("proc main\n{\n^rsi = xmm0\nxmm1 = ^rsi\n}\n");
+	struct Program program;
+	check(context, parse_program(&lexer, &program));
+
+	FILE* out = tmpfile();
+	generate_nasm(&program, out);
+	fflush(out);
+	rewind(out);
+
+	char buffer[1024];
+	size_t read = fread(buffer, 1, sizeof(buffer) - 1, out);
+	buffer[read] = '\0';
+	fclose(out);
+
+	check(context, strstr(buffer, "movsd [rsi], xmm0") != NULL); // float store
+	check(context, strstr(buffer, "movsd xmm1, [rsi]") != NULL); // float load
+	check(context, strstr(buffer, "; TODO") == NULL);
+
+	free_program(&program);
+}
+
 void run_codegen_tests(struct TestContext* context)
 {
+	test_generate_floats(context);
+	test_generate_float_compare(context);
+	test_generate_float_memory(context);
+	test_generate_add_zero_peephole(context);
 	test_generate_enum_struct(context);
 	test_generate_load(context);
 	test_generate_logical_registers(context);
diff --git a/tests/lexer_test.c b/tests/lexer_test.c
index 38644d9..a44b1f0 100644
--- a/tests/lexer_test.c
+++ b/tests/lexer_test.c
@@ -28,6 +28,14 @@ static void test_number_bases(struct TestContext* context)
 	check(context, token_matches(scan_token(&lexer), TOKEN_INTEGER, "0b1010"));
 }
 
+static void test_float_literals(struct TestContext* context)
+{
+	struct Lexer lexer = create_lexer("3.14 1.0 42");
+	check(context, token_matches(scan_token(&lexer), TOKEN_FLOAT, "3.14"));
+	check(context, token_matches(scan_token(&lexer), TOKEN_FLOAT, "1.0"));
+	check(context, scan_token(&lexer).type == TOKEN_INTEGER);
+}
+
 static void test_operators(struct TestContext* context)
 {
 	struct Lexer lexer = create_lexer("= == += != /");
@@ -92,6 +100,7 @@ void run_lexer_tests(struct TestContext* context)
 {
 	test_identifiers_and_integers(context);
 	test_number_bases(context);
+	test_float_literals(context);
 	test_operators(context);
 	test_literals(context);
 	test_keywords(context);
diff --git a/tests/parser_test.c b/tests/parser_test.c
index 71e1433..dbb250f 100644
--- a/tests/parser_test.c
+++ b/tests/parser_test.c
@@ -199,7 +199,7 @@ static void test_parse_stack(struct TestContext* context)
 	struct Statement statement = program.procs[0].body[0];
 	check(context, statement.kind == STATEMENT_STACK);
 	check(context, text_is(statement.stack.name, "buffer"));
-	check(context, text_is(statement.stack.size, "32"));
+	check(context, primary_is(statement.stack.size, "32"));
 
 	free_program(&program);
 }
diff --git a/tests/sema_test.c b/tests/sema_test.c
index e89f608..dc01d47 100644
--- a/tests/sema_test.c
+++ b/tests/sema_test.c
@@ -97,6 +97,14 @@ static void test_deref_needs_register(struct TestContext* context)
 	check(context, analyze_source("proc main\n{\nrax = ^rsi\n}\n"));
 }
 
+static void test_stack_size_constant(struct TestContext* context)
+{
+	check(context, analyze_source(
+		"struct P\n{\nx\ny\n}\nconst N = 4\n"
+		"proc main\n{\nstack a[P.size]\nstack b[N * 2]\nsyscall\n}\n"));
+	check(context, !analyze_source("proc main\n{\nstack a[rax]\n}\n"));
+}
+
 static void test_enum_struct_members(struct TestContext* context)
 {
 	check(context, analyze_source(
@@ -144,6 +152,7 @@ void run_sema_tests(struct TestContext* context)
 	test_const_expr_rejects_register(context);
 	test_const_expr_rejects_data(context);
 	test_deref_needs_register(context);
+	test_stack_size_constant(context);
 	test_enum_struct_members(context);
 	test_references_resolve(context);
 }