fix pkg tests

bait-lang · Jan 22, 2025 · 3451883 · 3451883
1 parent 6863358
commit 3451883
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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)