Implement the add-assign operator for VertexIter (#151)

apache · May 6, 2023 · 807f2da · 807f2da
1 parent b519b74
commit 807f2da
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/cpp/include/gar/graph.h b/cpp/include/gar/graph.h
@@ -213,6 +213,12 @@ class VertexIter {
     return ret;
   }
 
+  /** The add-assign operator. */
+  VertexIter& operator+=(IdType offset) {
+    cur_offset_ += offset;
+    return *this;
+  }
+
   /** The equality operator. */
   bool operator==(const VertexIter& rhs) const noexcept {
     return cur_offset_ == rhs.cur_offset_;

diff --git a/cpp/test/test_graph.cc b/cpp/test/test_graph.cc
@@ -56,6 +56,17 @@ TEST_CASE("test_vertices_collection") {
             << ", id=" << it_last.property<int64_t>("id").value()
             << ", firstName="
             << it_last.property<std::string>("firstName").value() << std::endl;
+  auto it_begin = it_last + (1 - count);
+
+  auto it = vertices.begin();
+  it += (count - 1);
+  REQUIRE(it.id() == it_last.id());
+  REQUIRE(it.property<int64_t>("id").value() ==
+          it_last.property<int64_t>("id").value());
+  it += (1 - count);
+  REQUIRE(it.id() == it_begin.id());
+  REQUIRE(it.property<int64_t>("id").value() ==
+          it_begin.property<int64_t>("id").value());
 }
 
 TEST_CASE("test_edges_collection", "[Slow]") {