boost_algorithm/test/partition_subrange_test.cpp

#include <boost/config.hpp>
#include <boost/algorithm/sort_subrange.hpp>
#include <boost/algorithm/cxx11/is_sorted.hpp>

#define BOOST_TEST_MAIN
#include <boost/test/unit_test.hpp>

#include <vector>
#include <iostream>

#if (__cplusplus >= 201103L) || defined(BOOST_NO_CXX98_RANDOM_SHUFFLE)
#include <random>

std::default_random_engine gen;
template<typename RandomIt>
void do_shuffle(RandomIt first, RandomIt last)
{ std::shuffle(first, last, gen); }
#else
template<typename RandomIt>
void do_shuffle(RandomIt first, RandomIt last)
{ std::random_shuffle(first, last); }
#endif

namespace ba = boost::algorithm;

template <typename Iter>
void check_sequence ( Iter first, Iter last, Iter sf, Iter sl )
{
// 	for (Iter i = first; i < last; ++i) {
// 		if (i != first) std::cout << ' ';
// 		if (i == sf) std::cout << ">";
// 		std::cout << *i;
// 		if (i == sl) std::cout << "<";
// 		}
// 	if (sl == last) std::cout << "<";
// 	std::cout << '\n';

	if (sf == sl) return;
	for (Iter i = first; i < sf; ++i)
		BOOST_CHECK(*i < *sf);
	for (Iter i = sf; i < sl; ++i) {
		if (first != sf) // if there is an element before the subrange
			BOOST_CHECK(*i > *(sf-1));
		if (last != sl) // if there is an element after the subrange
			BOOST_CHECK(*i < *sl);
		}
	for (Iter i = sl; i < last; ++i)
		BOOST_CHECK(*(sl-1) < *i);
}

template <typename Iter, typename Pred>
void check_sequence ( Iter first, Iter last, Iter sf, Iter sl, Pred p )
{
	if (sf == sl) return;
	for (Iter i = first; i < sf; ++i)
		BOOST_CHECK(p(*i, *sf));
	for (Iter i = sf; i < sl; ++i) {
		if (first != sf) // if there is an element before the subrange
			BOOST_CHECK(p(*(sf-1), *i));
		if (last != sl) // if there is an element after the subrange
			BOOST_CHECK(p(*i, *sl));
		}
	for (Iter i = sl; i < last; ++i)
		BOOST_CHECK(p(*(sl-1), *i));

}

// 	for ( int i = 0; i < v.size(); ++i )
// 		std::cout << v[i] << ' ';
// 	std::cout << std::endl;


BOOST_AUTO_TEST_CASE( test_main )
{
	{
	std::vector<int> v;
	for ( int i = 0; i < 10; ++i )
		v.push_back(i);

	const std::vector<int>::iterator b = v.begin();
	ba::partition_subrange(b, v.end(), b + 3, b + 6);
	check_sequence        (b, v.end(), b + 3, b + 6);

// 	BOOST_CHECK_EQUAL(v[3], 3);
// 	BOOST_CHECK_EQUAL(v[4], 4);
// 	BOOST_CHECK_EQUAL(v[5], 5);

//	Mix them up and try again - single element
	do_shuffle(v.begin(), v.end());
	ba::partition_subrange(b, v.end(), b + 7, b + 8);
	check_sequence        (b, v.end(), b + 7, b + 8);

// 	BOOST_CHECK_EQUAL(v[7], 7);

//	Mix them up and try again - at the end
	do_shuffle(v.begin(), v.end());
	ba::partition_subrange(b, v.end(), b + 7, v.end());
	check_sequence        (b, v.end(), b + 7, v.end());

// 	BOOST_CHECK_EQUAL(v[7], 7);
// 	BOOST_CHECK_EQUAL(v[8], 8);
// 	BOOST_CHECK_EQUAL(v[9], 9);

//	Mix them up and try again - at the beginning
	do_shuffle(v.begin(), v.end());
	ba::partition_subrange(b, v.end(), b, b + 2);
	check_sequence        (b, v.end(), b, b + 2);

// 	BOOST_CHECK_EQUAL(v[0], 0);
// 	BOOST_CHECK_EQUAL(v[1], 1);

//	Mix them up and try again - empty subrange
	do_shuffle(v.begin(), v.end());
	ba::partition_subrange(b, v.end(), b, b);
	check_sequence        (b, v.end(), b, b);

//	Mix them up and try again - entire subrange
	do_shuffle(v.begin(), v.end());
	ba::partition_subrange(b, v.end(), b, v.end());
	check_sequence        (b, v.end(), b, v.end());
	}

	{
	std::vector<int> v;
	for ( int i = 0; i < 10; ++i )
		v.push_back(i);

	const std::vector<int>::iterator b = v.begin();
	ba::partition_subrange(b, v.end(), b + 3, b + 6, std::greater<int>());
	check_sequence        (b, v.end(), b + 3, b + 6, std::greater<int>());

// 	BOOST_CHECK_EQUAL(v[3], 6);
// 	BOOST_CHECK_EQUAL(v[4], 5);
// 	BOOST_CHECK_EQUAL(v[5], 4);

//	Mix them up and try again - single element
	do_shuffle(v.begin(), v.end());
	ba::partition_subrange(b, v.end(), b + 7, b + 8, std::greater<int>());
	check_sequence        (b, v.end(), b + 7, b + 8, std::greater<int>());

// 	BOOST_CHECK_EQUAL(v[7], 2);

//	Mix them up and try again - at the end
	do_shuffle(v.begin(), v.end());
	ba::partition_subrange(b, v.end(), b + 7, v.end(), std::greater<int>());
	check_sequence        (b, v.end(), b + 7, v.end(), std::greater<int>());

// 	BOOST_CHECK_EQUAL(v[7], 2);
// 	BOOST_CHECK_EQUAL(v[8], 1);
// 	BOOST_CHECK_EQUAL(v[9], 0);

//	Mix them up and try again - at the beginning
	do_shuffle(v.begin(), v.end());
	ba::partition_subrange(b, v.end(), b, b + 2, std::greater<int>());
	check_sequence        (b, v.end(), b, b + 2, std::greater<int>());

// 	BOOST_CHECK_EQUAL(v[0], 9);
// 	BOOST_CHECK_EQUAL(v[1], 8);

//	Mix them up and try again - empty subrange
	do_shuffle(v.begin(), v.end());
	ba::partition_subrange(b, v.end(), b, b, std::greater<int>());
	check_sequence        (b, v.end(), b, b, std::greater<int>());

//	Mix them up and try again - entire subrange
	do_shuffle(v.begin(), v.end());
	ba::partition_subrange(b, v.end(), b, v.end(), std::greater<int>());
	check_sequence        (b, v.end(), b, v.end(), std::greater<int>());
	}
}