From 034cd404a129103a8dd7747e6bd00ffd5550da93 Mon Sep 17 00:00:00 2001
From: Lexi Winter <lexi@le-fay.org>
Date: Mon, 30 Jun 2025 07:51:23 +0100
Subject: refactoring

---
 nihil.generator/generator.test.cc | 56 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 56 insertions(+)
 create mode 100644 nihil.generator/generator.test.cc

(limited to 'nihil.generator/generator.test.cc')

diff --git a/nihil.generator/generator.test.cc b/nihil.generator/generator.test.cc
new file mode 100644
index 0000000..49272b4
--- /dev/null
+++ b/nihil.generator/generator.test.cc
@@ -0,0 +1,56 @@
+/*
+ * This source code is released into the public domain.
+ */
+
+#include <coroutine>
+#include <ranges>
+#include <vector>
+
+#include <catch2/catch_test_macros.hpp>
+
+import nihil.generator;
+
+TEST_CASE("generator: basic", "[generator]") 
+{
+	auto fn = [] () -> nihil::generator<int> {
+		co_yield 1;
+		co_yield 2;
+		co_yield 3;
+	};
+
+	auto values = std::vector<int>();
+	std::ranges::copy(fn(), std::back_inserter(values));
+
+	REQUIRE(values == std::vector{1, 2, 3});
+}
+
+TEST_CASE("generator: exceptions", "[generator]") 
+{
+	auto fn = [] () -> nihil::generator<int> {
+		co_yield 1;
+		throw std::runtime_error("test");
+	};
+
+	auto range = fn();
+	auto it = std::ranges::begin(range);
+	REQUIRE(*it == 1);
+	REQUIRE_THROWS_AS(it++, std::runtime_error);
+}
+
+TEST_CASE("generator: elements_of", "[generator]") 
+{
+	auto fn1 = [] -> nihil::generator<int> {
+		co_yield 1;
+		co_yield 2;
+		co_yield 3;
+	};
+
+	auto fn2 = [&fn1] -> nihil::generator<int> {
+		co_yield nihil::elements_of(fn1());
+	};
+
+	auto values = std::vector<int>();
+	std::ranges::copy(fn2(), std::back_inserter(values));
+
+	REQUIRE(values == std::vector{1, 2, 3});
+}
-- 
cgit v1.2.3