From 345188318ae7965e7a622e53e901dffc7eeffd29 Mon Sep 17 00:00:00 2001
From: Lukas Neubert <40118727+serkonda7@users.noreply.github.com>
Date: Wed, 22 Jan 2025 15:18:02 +0100
Subject: [PATCH] fix pkg tests

---
 lib/bait/gen/c/fun.bt  | 2 +-
 lib/bait/gen/js/fun.bt | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/lib/bait/gen/c/fun.bt b/lib/bait/gen/c/fun.bt
index bba97a43..d0f7d763 100644
--- a/lib/bait/gen/c/fun.bt
+++ b/lib/bait/gen/c/fun.bt
@@ -44,7 +44,7 @@ fun (mut g Gen) fun_decl(node ast.FunDecl) {
 		name = c_name(sym.name + '_' + node.name)
 	} else {
 		// TODO remove exception for main and builtin
-		if g.pkg == "main" or g.pkg == "builtin" {
+		if g.pkg == "main" or g.pkg == "builtin" or node.is_test {
 			name = c_esc(node.name)
 		} else {
 			name = c_name(g.pkg + "." + node.name)
diff --git a/lib/bait/gen/js/fun.bt b/lib/bait/gen/js/fun.bt
index 5c76890a..408288ec 100644
--- a/lib/bait/gen/js/fun.bt
+++ b/lib/bait/gen/js/fun.bt
@@ -27,7 +27,7 @@ fun (mut g Gen) fun_decl(node ast.FunDecl) {
 		name = js_esc(sym.name + '_' + node.name)
 	} else {
 		// TODO remove exception for main and builtin
-		if g.pkg == "main" or g.pkg == "builtin" {
+		if g.pkg == "main" or g.pkg == "builtin" or node.is_test {
 			name = js_esc(node.name)
 		} else {
 			name = js_name(g.pkg + "." + node.name)