fix: prevent invalid types in test/fuzz functions

compiler/noirc_frontend/src/elaborator/mod.rs

@@ -929,6 +929,8 @@ impl<'context> Elaborator<'context> {
         let name_ident = HirIdent::non_trait_method(id, location);
 
         let is_entry_point = self.is_entry_point_function(func, in_contract);
+        let is_test_or_fuzz =
+            func.attributes().is_test_function() || func.attributes().is_fuzzing_harness();
 
         // Both the #[fold] and #[no_predicates] alter a function's inline type and code generation in similar ways.
         // In certain cases such as type checking (for which the following flag will be used) both attributes
@@ -993,12 +995,12 @@ impl<'context> Elaborator<'context> {
 
             self.check_if_type_is_valid_for_program_input(
                 &typ,
-                is_entry_point,
+                is_entry_point || is_test_or_fuzz,
                 has_inline_attribute,
                 type_location,
             );
 
-            if is_entry_point {
+            if is_entry_point || is_test_or_fuzz {
                 self.mark_type_as_used(&typ);
             }
 

compiler/noirc_frontend/src/hir/type_check/errors.rs

@@ -618,7 +618,7 @@ impl<'a> From<&'a TypeCheckError> for Diagnostic {
             }
             TypeCheckError::InvalidTypeForEntryPoint { location } => Diagnostic::simple_error(
                 "Only sized types may be used in the entry point to a program".to_string(),
-                "Slices, references, or any type containing them may not be used in main, contract functions, or foldable functions".to_string(), *location),
+                "Slices, references, or any type containing them may not be used in main, contract functions, test functions, fuzz functions or foldable functions".to_string(), *location),
             TypeCheckError::MismatchTraitImplNumParameters {
                 expected_num_parameters,
                 actual_num_parameters,

test_programs/compile_failure/test_invalid_argument_type/Nargo.toml

@@ -0,0 +1,7 @@
+[package]
+name = "test_invalid_argument_type"
+type = "bin"
+authors = [""]
+compiler_version = ">=0.33.0"
+
+[dependencies]

test_programs/compile_failure/test_invalid_argument_type/src/main.nr

@@ -0,0 +1,4 @@
+fn main() {}
+
+#[test]
+fn test(_arg: &mut i32) {}