Add Sequence<AdditiveArithmetic>.sum()

RevenueCat · Jun 22, 2023 · 100b824 · 100b824
1 parent bb1b3e4
commit 100b824
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 0 deletions.
diff --git a/Sources/FoundationExtensions/Array+Extensions.swift b/Sources/FoundationExtensions/Array+Extensions.swift
@@ -36,3 +36,12 @@ extension Collection {
     }
 
 }
+
+extension Sequence where Element: AdditiveArithmetic {
+
+    /// - Returns: the sum of the elements in the sequence.
+    func sum() -> Element {
+        return reduce(.zero, +)
+    }
+
+}
diff --git a/Tests/UnitTests/FoundationExtensions/ArrayExtensionsTests.swift b/Tests/UnitTests/FoundationExtensions/ArrayExtensionsTests.swift
@@ -57,4 +57,18 @@ class ArrayExtensionsTests: TestCase {
         expect([1].onlyElement) == 1
     }
 
+    // MARK: - sum
+
+    func testSumEmptyArray() {
+        expect([Int]().sum()) == 0
+    }
+
+    func testSumSingleElementArray() {
+        expect([5].sum()) == 5
+    }
+
+    func testSumArrayWithMultipleElements() {
+        expect([1, 2, 3].sum()) == 6
+    }
+
 }