#include "catch.hpp"
#include "enumerate.hh"

#include "narrow.hh"

#include <array>
#include <type_traits>
#include <vector>

TEST_CASE("enumerate: basic")
{
	std::array in = {'a', 'd', 'z'};
	std::vector<size_t> out1;
	std::vector<char> out2;

	for (auto [i, e] : enumerate(in)) {
		static_assert(std::is_same_v<decltype(i), const size_t&>);
		static_assert(std::is_same_v<decltype(e), char&>);
		out1.push_back(i);
		out2.push_back(e);
	}

	CHECK(out1 == std::vector<size_t>{0, 1, 2});
	CHECK(out2 == std::vector{'a', 'd', 'z'});
}

TEST_CASE("enumerate: transform")
{
	std::vector in{3, 9, 11};
	std::vector<size_t> out;

	for (auto [i, e] : enumerate(in)) {
		static_assert(std::is_same_v<decltype(i), const size_t&>);
		static_assert(std::is_same_v<decltype(e), int&>);
		out.push_back(i);
		e = narrow_cast<int>(2 * e + i);
	}

	CHECK(in == std::vector{6, 19, 24});
	CHECK(out == std::vector<size_t>{0, 1, 2});
}