feat(std): abs function

amber-lang · b1ek · Aug 4, 2024 · Jul 31, 2024 · Jul 31, 2024 · Jul 31, 2024
commit 87b936951a5b5d5df9f8fc1f5325ac820af5cd80
diff --git a/src/std/math.ab b/src/std/math.ab
@@ -3,17 +3,27 @@ pub fun sum(list: [Num]): Num {
     return unsafe $echo "{list}" | awk '\{s=0; for (i=1; i<=NF; i++) s+=\$i; print s}'$ as Num
 }
 
+/// Returns the number rounded to the nearest integer
 #[allow_absurd_cast]
 pub fun round(number: Num): Num {
     return unsafe $printf "%0.f" {number}$ as Num
 }
 
+/// Returns the largest integer less than or equal to the number
 #[allow_absurd_cast]
 pub fun floor(number: Num): Num {
-    return unsafe $echo {number} | awk '\{printf "%d", (\$1 < 0 ? int(\$1) - 1 : int(\$1))}'$  as Num
+    return unsafe $echo {number} | awk '\{printf "%d", (\$1 < 0 ? int(\$1) - 1 : int(\$1))}'$ as Num
 }
 
+/// Returns the smallest integer greater than or equal to the number
 #[allow_absurd_cast]
 pub fun ceil(number: Num): Num {
     return floor(number) + 1
 }
+
+/// Returns the absolute value of the number
+#[allow_absurd_cast]
+pub fun abs(number: Num): Num {
+    if number < 0: return -number
+    return number
+}
diff --git a/src/tests/stdlib/abs.ab b/src/tests/stdlib/abs.ab
@@ -0,0 +1,10 @@
+import * from "std/math"
+
+// Output 
+// 1
+// 1
+
+main {
+    echo abs(1);
+    echo abs(-1);
+}