fn check_return_statements( statement: &Statement, expected_count: usize, function_name: &str, return_arities: &HashMap<(String, usize), usize>, diagnostics: &mut Vec<Diagnostic>, )