#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});

}