lib: add iterate methods on Collection

[nit.git] / tests / test_iterate.nit

diff --git a/tests/test_iterate.nit b/tests/test_iterate.nit
new file mode 100644 (file)
index 0000000..b2d95c5
--- /dev/null
+++ b/tests/test_iterate.nit
@@ -0,0 +1,82 @@
+# This file is part of NIT ( http://www.nitlanguage.org ).
+#
+# Copyright 2009 Jean Privat <jean@pryen.org>
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+fun test_iterator(c: Collection[Int])
+do
+       var idx = 0
+       var it = c.iterator
+       while it.is_ok do
+               var i = it.item
+               print "  {idx}->{i}"
+               it.next
+               idx += 1
+       end
+end
+
+fun test_iterate(c: Collection[Int])
+do
+       var idx = 0
+       c.iterate !each i do
+               print "  {idx}->{i}"
+               idx += 1
+       end
+end
+
+fun test_for(c: Collection[Int])
+do
+       var idx = 0
+       for i in c do
+               print "  {idx}->{i}"
+               idx += 1
+       end
+end
+
+fun test_coll(c: Collection[Int], s: String)
+do
+       print("{s}: iterator")
+       test_iterator(c)
+       print("{s}: iterate")
+       test_iterate(c)
+       print("{s}: for")
+       test_for(c)
+       print("")
+end
+
+fun init_seq(c: SimpleCollection[Int]): SimpleCollection[Int]
+do
+       for i in [0..5[ do
+               c.add(i)
+       end
+       return c
+end
+
+fun init_map(c: Map[Int, Int]): Map[Int, Int]
+do
+       for i in [0..5[ do
+               c[i*10] = i
+       end
+       return c
+end
+
+test_coll(init_seq(new Array[Int]), "Array")
+test_coll(init_seq(new List[Int]), "List")
+test_coll(init_seq(new ArraySet[Int]), "ArraySet")
+test_coll(init_seq(new HashSet[Int]), "HashSet")
+test_coll([0..5[, "ORange")
+test_coll([0..4], "CRange")
+test_coll(init_map(new ArrayMap[Int, Int]), "ArrayMap")
+test_coll(init_map(new HashMap[Int, Int]), "HashMap")
+