Skip to content

Commit 8d2d9c5

Browse files
committed
fix(math): prime? and divs should only accept integers, closes #49
1 parent 51974c1 commit 8d2d9c5

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

Math.ark

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -255,7 +255,7 @@
255255
(let prime? (fun (n)
256256
(if (= 2 n)
257257
true
258-
(if (or (= 0 (mod n 2)) (= 1 n))
258+
(if (or (= 0 (mod n 2)) (= 1 n) (!= 0 (mod n 1)))
259259
false
260260
{
261261
(let k (ceil (+ 1 (sqrt n))))
@@ -276,6 +276,7 @@
276276
# @author https://github.com/Wafelack
277277
(let divs (fun (n) {
278278
(assert (>= n 1) "divs: n must be greater or equal to 1")
279+
(assert (= 0 (mod n 1)) "divs: n must be an integer")
279280
(mut i 1)
280281
(mut divisors [])
281282
(let top (ceil (/ n 2)))

0 commit comments

Comments
 (0)