75 lines
2.2 KiB
C++
75 lines
2.2 KiB
C++
|
// (C) Copyright Jeremy Siek 2002.
|
||
|
// Distributed under the Boost Software License, Version 1.0. (See
|
||
|
// accompanying file LICENSE_1_0.txt or copy at
|
||
|
// http://www.boost.org/LICENSE_1_0.txt)
|
||
|
|
||
|
#include <boost/pending/disjoint_sets.hpp>
|
||
|
#include <boost/core/lightweight_test.hpp>
|
||
|
#include <boost/cstdlib.hpp>
|
||
|
|
||
|
template < typename DisjointSet > struct test_disjoint_set
|
||
|
{
|
||
|
static void do_test()
|
||
|
{
|
||
|
// The following tests are pretty lame, just a basic sanity check.
|
||
|
// Industrial strength tests still need to be written.
|
||
|
|
||
|
#if !defined(__MWERKS__) || __MWERKS__ > 0x3003
|
||
|
std::size_t elts[]
|
||
|
#else
|
||
|
std::size_t elts[4]
|
||
|
#endif
|
||
|
= { 0, 1, 2, 3 };
|
||
|
|
||
|
const int N = sizeof(elts) / sizeof(*elts);
|
||
|
|
||
|
DisjointSet ds(N);
|
||
|
|
||
|
ds.make_set(elts[0]);
|
||
|
ds.make_set(elts[1]);
|
||
|
ds.make_set(elts[2]);
|
||
|
ds.make_set(elts[3]);
|
||
|
|
||
|
BOOST_TEST(ds.find_set(0) != ds.find_set(1));
|
||
|
BOOST_TEST(ds.find_set(0) != ds.find_set(2));
|
||
|
BOOST_TEST(ds.find_set(0) != ds.find_set(3));
|
||
|
BOOST_TEST(ds.find_set(1) != ds.find_set(2));
|
||
|
BOOST_TEST(ds.find_set(1) != ds.find_set(3));
|
||
|
BOOST_TEST(ds.find_set(2) != ds.find_set(3));
|
||
|
|
||
|
ds.union_set(0, 1);
|
||
|
ds.union_set(2, 3);
|
||
|
BOOST_TEST(ds.find_set(0) != ds.find_set(3));
|
||
|
int a = ds.find_set(0);
|
||
|
BOOST_TEST(a == ds.find_set(1));
|
||
|
int b = ds.find_set(2);
|
||
|
BOOST_TEST(b == ds.find_set(3));
|
||
|
|
||
|
ds.link(a, b);
|
||
|
BOOST_TEST(ds.find_set(a) == ds.find_set(b));
|
||
|
BOOST_TEST(1 == ds.count_sets(elts, elts + N));
|
||
|
|
||
|
ds.normalize_sets(elts, elts + N);
|
||
|
ds.compress_sets(elts, elts + N);
|
||
|
BOOST_TEST(1 == ds.count_sets(elts, elts + N));
|
||
|
}
|
||
|
};
|
||
|
|
||
|
int main(int, char*[])
|
||
|
{
|
||
|
using namespace boost;
|
||
|
{
|
||
|
typedef disjoint_sets_with_storage< identity_property_map,
|
||
|
identity_property_map, find_with_path_halving >
|
||
|
ds_type;
|
||
|
test_disjoint_set< ds_type >::do_test();
|
||
|
}
|
||
|
{
|
||
|
typedef disjoint_sets_with_storage< identity_property_map,
|
||
|
identity_property_map, find_with_full_path_compression >
|
||
|
ds_type;
|
||
|
test_disjoint_set< ds_type >::do_test();
|
||
|
}
|
||
|
return boost::report_errors();
|
||
|
}
|