From dbe998618dc32600dbca8a34b0246616fb4a5b69 Mon Sep 17 00:00:00 2001
From: Alon Krymgand <me@alon.kr>
Date: Sat, 8 Feb 2025 12:23:13 +0200
Subject: [PATCH] Fixed tests to pass

---
 gen/function_generator_test.go    | 6 +++---
 gen/instruction_generator_test.go | 4 ++--
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/gen/function_generator_test.go b/gen/function_generator_test.go
index b71f58c..268775c 100644
--- a/gen/function_generator_test.go
+++ b/gen/function_generator_test.go
@@ -97,10 +97,10 @@ func TestIfElseFunctionGeneration(t *testing.T) {
 	src := `func @toBool $32 %n {
 	JZ %n .zero
 .nonzero
-	%bool = ADD $32 #1 $32 #0
+	$32 %bool = ADD $32 #1 $32 #0
 	JMP .end
 .zero
-	%bool = ADD $32 #0 $32 #0
+	$32 %bool = ADD $32 #0 $32 #0
 .end
 	RET
 }`
@@ -135,7 +135,7 @@ func TestEmptyFunctionGeneration(t *testing.T) {
 
 func TestNoReturnFunctionGeneration(t *testing.T) {
 	src := `func @noReturn {
-				%n = ADD $32 #1 $32 #2
+				$32 %n = ADD $32 #1 $32 #2
 			}`
 	function, results := generateFunctionFromSource(t, src)
 	assert.False(t, results.IsEmpty())
diff --git a/gen/instruction_generator_test.go b/gen/instruction_generator_test.go
index 1de9086..7ffdffc 100644
--- a/gen/instruction_generator_test.go
+++ b/gen/instruction_generator_test.go
@@ -183,7 +183,7 @@ func (m *InstructionMap) GetInstructionDefinition(
 }
 
 func TestInstructionCreateTarget(t *testing.T) {
-	src := core.NewSourceView("%c = ADD %a %b\n")
+	src := core.NewSourceView("$32 %c = ADD %a %b\n")
 	tkns, err := lex.NewTokenizer().Tokenize(src)
 	assert.NoError(t, err)
 
@@ -221,5 +221,5 @@ func TestInstructionCreateTarget(t *testing.T) {
 	assert.NotNil(t, target)
 	assert.Equal(t, "%c", target.Name)
 	assert.Equal(t, intType, target.Type.Base)
-	assert.Equal(t, src.Unmanaged().Subview(0, 2), target.Declaration)
+	assert.Equal(t, src.Unmanaged().Subview(0, 6), target.Declaration)
 }