Apply code review

exercism · Jan 10, 2024 · 721f401 · 721f401
1 parent 9112b97
commit 721f401
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/exercises/practice/square-root/.meta/example.rkt b/exercises/practice/square-root/.meta/example.rkt
@@ -2,7 +2,9 @@
 
 (provide square-root)
 
-(define (square-root radicand)
-    (for/first ([num (in-naturals)]
-                #:when (equal? (expt num 2) radicand))
-      num))
+(define/contract (square-root radicand)
+  (-> exact-positive-integer? (or/c exact-positive-integer?
+                                    false?))
+  (for/first ([num (in-inclusive-range 1 radicand)]
+              #:when (equal? (expt num 2) radicand))
+    num))