diff options
Diffstat (limited to 'tests/run/iterators.scala')
-rw-r--r-- | tests/run/iterators.scala | 136 |
1 files changed, 136 insertions, 0 deletions
diff --git a/tests/run/iterators.scala b/tests/run/iterators.scala new file mode 100644 index 000000000..e2a97ec7a --- /dev/null +++ b/tests/run/iterators.scala @@ -0,0 +1,136 @@ +//############################################################################ +// Iterators +//############################################################################ + +//############################################################################ + +import scala.language.postfixOps + +object Test { + + def check_from: Int = { + val it1 = Iterator.from(-1) + val it2 = Iterator.from(0, -1) + it1.next + it2.next + } + + def check_range: Int = { + val xs1 = Iterator.range(0, 10, 2) toList; + val xs2 = Iterator.range(0, 10, -2) toList; + val xs3 = Iterator.range(10, 0, -2) toList; + val xs4 = Iterator.range(10, 0, 2) toList; + val xs5 = Iterator.range(0, 10, 11) toList; + xs1.length + xs2.length + xs3.length + xs4.length + xs5.length + } + + def check_range2: Int = { + val r1start = 0 + val r1end = 10 + val r1step = 1 + val r1 = Iterator.range(r1start, r1end, r1step) toList; + val r2 = Iterator.range(r1start, r1end, r1step + 1) toList; + val r3 = Iterator.range(r1end, r1start, -r1step) toList; + val r4 = Iterator.range(0, 10, 11) toList; + // 10 + 5 + 10 + 1 + r1.length + r2.length + r3.length + r4.length + } + + def check_range3: Int = { + def trues(xs: List[Boolean]) = xs.foldLeft(0)((a, b) => if (b) a+1 else a) + val r1 = Iterator.range(0, 10) + val xs1 = List(r1 contains 5, r1 contains 6) + val r2a = Iterator.range(0, 10, 2) + val r2b = Iterator.range(0, 10, 2) + val xs2 = List(r2a contains 5, r2b contains 6) + val r3 = Iterator.range(0, 10, 11) + val xs3 = List(r3 contains 5, r3 contains 6) + // 2 + 1 + 0 + trues(xs1) + trues(xs2) + trues(xs3) + } + + def check_take: Int = { + val it1 = Iterator.from(0) + val xs1 = it1 take 10 toList; + xs1.length + } + + def check_drop: Int = { + val it1 = Iterator.from(0) + val it2 = it1 map { 2 * _ } + val n1 = it1 drop 2 next + val n2 = it2 drop 2 next; + n1 + n2 + } + + def check_foreach: Int = { + val it1 = Iterator.from(0) take 20 + var n = 0 + it1 foreach { n += _ } + n + } + + def check_forall: Int = { + val it1 = Iterator.from(0) + val it2 = Iterator.from(1) + 0 + } + + def check_fromArray: Int = { // ticket #429 + val a = List(1, 2, 3, 4).toArray + var xs0 = a.iterator.toList; + var xs1 = a.slice(0, 1).iterator.toList; + var xs2 = a.slice(0, 2).iterator.toList; + var xs3 = a.slice(0, 3).iterator.toList; + var xs4 = a.slice(0, 4).iterator.toList; + xs0.length + xs1.length + xs2.length + xs3.length + xs4.length + } + + def check_toSeq: String = + List(1, 2, 3, 4, 5).iterator.toSeq.mkString("x") + + def check_indexOf: String = { + val i = List(1, 2, 3, 4, 5).indexOf(4) + val j = List(1, 2, 3, 4, 5).indexOf(16) + "" + i + "x" + j + } + + def check_findIndexOf: String = { + val i = List(1, 2, 3, 4, 5).indexWhere { x: Int => x >= 4 } + val j = List(1, 2, 3, 4, 5).indexWhere { x: Int => x >= 16 } + "" + i + "x" + j + } + + def check_success[A](name: String, closure: => A, expected: A): Unit = { + print("test " + name) + try { + val actual: A = closure + if (actual == expected) + print(" was successful") + else + print(" failed: expected "+ expected +", found "+ actual) + } + catch { + case exception: Throwable => + print(" raised exception " + exception) + } + println() + } + + def main(args: Array[String]): Unit = { + check_success("check_from", check_from, -1) + check_success("check_range", check_range, 11) + check_success("check_range2", check_range2, 26) + check_success("check_range3", check_range3, 3) + check_success("check_take", check_take, 10) + check_success("check_drop", check_drop, 12) + check_success("check_foreach", check_foreach, 190) + check_success("check_forall", check_forall, 0) + check_success("check_fromArray",check_fromArray, 14) + check_success("check_toSeq", check_toSeq, "1x2x3x4x5") + check_success("check_indexOf", check_indexOf, "3x-1") + check_success("check_findIndexOf", check_findIndexOf, "3x-1") + println() + } +} + +//############################################################################ |