Provide average() method for iterables, sequences and arrays.

#KT-3843 Fixed
2015-04-16 16:12:05 +03:00
parent f402ae8540
commit 5dacb5a745
4 changed files with 495 additions and 0 deletions
@@ -438,6 +438,16 @@ class CollectionTest {
        expect(3.0.toFloat()) { sequenceOf<Float>(1.0.toFloat(), 2.0.toFloat()).sum() }
    }

+    test fun average() {
+        expect(0.0) { arrayListOf<Int>().average() }
+        expect(3.8) { listOf(1, 2, 5, 8, 3).average() }
+        expect(2.1) { sequenceOf(1.6, 2.6, 3.6, 0.6).average() }
+        expect(100.0) { arrayListOf<Byte>(100, 100, 100, 100, 100, 100).average() }
+        val n = 100
+        val range = 0..n
+        expect(n.toDouble()/2) { range.average() }
+    }
+
    test fun takeReturnsFirstNElements() {
        expect(listOf(1, 2, 3, 4, 5)) { (1..10) take 5 }
        expect(listOf(1, 2, 3, 4, 5)) { (1..10).toList().take(5) }