Aggregate to containers, including tests

Author: jay-tux

inc/aggregators.hpp

@@ -0,0 +1,33 @@
+#ifndef _FPGEN_AGGREGATORS
+#define _FPGEN_AGGREGATORS
+
+#include "generator.hpp"
+#include <forward_list>
+#include <tuple>
+#include <type_traits>
+
+namespace fpgen {
+template <typename T, typename... Args,
+          template <typename...> typename Container>
+Container<T, Args...> &aggregate_to(generator<T> &gen,
+                                    Container<T, Args...> &out) {
+  while (gen) {
+    out.push_back(gen());
+  }
+  return out;
+}
+
+template <typename TKey, typename TVal, typename... Args,
+          template <typename...> typename Container>
+Container<TKey, TVal, Args...> &
+tup_aggregate_to(generator<std::tuple<TKey, TVal>> &gen,
+                 Container<TKey, TVal, Args...> &out) {
+  while (gen) {
+    std::tuple<TKey, TVal> tup = gen();
+    out[std::get<0>(tup)] = std::get<1>(tup);
+  }
+  return out;
+}
+} // namespace fpgen
+
+#endif

test/Makefile

@@ -1,6 +1,6 @@
 SOURCES=$(shell find $(SRCD) -name '*.cpp')
 DEPS=$(SOURCES:$(SRCD)/%.cpp=$(OBJD)/%.d)
-TESTS=generator sources manip
+TESTS=generator sources manip aggreg
 TESTOBJ=$(TESTS:%=$(OBJD)/test_%.o)
 
 CONAN_PKG_OVERRIDE=gtest

test/src/test_aggreg.cpp

@@ -0,0 +1,69 @@
+#include "aggregators.hpp"
+#include "generator.hpp"
+#include "manipulators.hpp"
+#include "gmock/gmock.h"
+#include "gtest/gtest.h"
+
+#include <map>
+#include <vector>
+
+fpgen::generator<size_t> a_empty() { co_return 0; }
+
+fpgen::generator<size_t> values() {
+  size_t i = 0;
+  size_t j = 1;
+  co_yield 0;
+  co_yield 1;
+
+  while (j < 21) {
+    size_t tmp = i + j;
+    co_yield tmp;
+    i = j;
+    j = tmp;
+  }
+
+  co_return i + j;
+}
+
+TEST(aggregate, empty) {
+  auto gen = a_empty();
+  gen();
+  std::vector<size_t> res;
+  EXPECT_EQ(0, fpgen::aggregate_to(gen, res).size());
+}
+
+TEST(aggregate, vector) {
+  auto gen = values();
+  std::vector<size_t> res;
+  fpgen::aggregate_to(gen, res);
+
+  EXPECT_EQ(0, res[0]);
+  EXPECT_EQ(1, res[1]);
+  EXPECT_EQ(1, res[2]);
+  EXPECT_EQ(2, res[3]);
+  EXPECT_EQ(3, res[4]);
+  EXPECT_EQ(5, res[5]);
+  EXPECT_EQ(8, res[6]);
+  EXPECT_EQ(13, res[7]);
+  EXPECT_EQ(21, res[8]);
+  EXPECT_EQ(34, res[9]);
+  EXPECT_EQ(res.size(), 10);
+}
+
+TEST(aggregate, map) {
+  fpgen::generator<size_t> sources[2] = {values(), values()};
+  auto gen = fpgen::zip(sources[0], sources[1]);
+  std::map<size_t, size_t> res;
+  fpgen::tup_aggregate_to(gen, res);
+
+  EXPECT_EQ(0, res[0]);
+  EXPECT_EQ(1, res[1]);
+  EXPECT_EQ(2, res[2]);
+  EXPECT_EQ(3, res[3]);
+  EXPECT_EQ(5, res[5]);
+  EXPECT_EQ(8, res[8]);
+  EXPECT_EQ(13, res[13]);
+  EXPECT_EQ(21, res[21]);
+  EXPECT_EQ(34, res[34]);
+  EXPECT_EQ(res.size(), 9);
+}