[22_1] make array iterable with range based for loop

XmacsLabs · May 23, 2024 · 2faf0a7 · 2faf0a7
1 parent 2e89509
commit 2faf0a7
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/Kernel/Containers/array.hpp b/Kernel/Containers/array.hpp
@@ -159,6 +159,9 @@ template <class T> class array {
    * @return   A reference to the element at the specified index.
    */
   inline T& operator[] (int i) { return rep->a[i]; }
+
+  T* begin () { return rep->a; }
+  T* end () { return rep->a + rep->n; }
 };
 CONCRETE_TEMPLATE_CODE (array, class, T);
 

diff --git a/tests/Kernel/Containers/array_test.cpp b/tests/Kernel/Containers/array_test.cpp
@@ -112,3 +112,19 @@ TEST_CASE ("array of string") {
   arr << "】";
   CHECK (contains (string ("】"), arr));
 }
+
+TEST_CASE ("test iteration") {
+  int expectedIndex= 0;
+  for (const auto element : five_elem) {
+    CHECK_EQ (element, five_elem[expectedIndex]);
+    ++expectedIndex;
+  }
+  CHECK_EQ (expectedIndex, 5);
+
+  expectedIndex= 0;
+  for (const auto element : zero_elem) {
+    CHECK_EQ (element, zero_elem[expectedIndex]);
+    ++expectedIndex;
+  }
+  CHECK_EQ (expectedIndex, 0);
+}