From 9ea786eda539780e89544d7d4da23d9430fcdcdb Mon Sep 17 00:00:00 2001
From: Rorik Star Platinum <kv-rust-dev@proton.me>
Date: Wed, 10 Dec 2025 18:12:24 +0300
Subject: [PATCH] grains is should panic

---
 rust/grains/src/lib.rs | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/rust/grains/src/lib.rs b/rust/grains/src/lib.rs
index ab7bb6c..1681d66 100644
--- a/rust/grains/src/lib.rs
+++ b/rust/grains/src/lib.rs
@@ -1,7 +1,10 @@
 pub fn square(s: u32) -> u64 {
+    if s == 0 || s > 64 {
+        panic!("Square must be between 1 and 64");
+    }
     2_u64.pow(s - 1)
 }
 
 pub fn total() -> u64 {
-    (1..=64).map(|x| square(x)).sum()
+    (1..=64).map(square).sum()
 }