williamr@2: //=======================================================================
williamr@2: // Copyright 2002 Indiana University.
williamr@2: // Authors: Andrew Lumsdaine, Lie-Quan Lee, Jeremy G. Siek
williamr@2: //
williamr@2: // Distributed under the Boost Software License, Version 1.0. (See
williamr@2: // accompanying file LICENSE_1_0.txt or copy at
williamr@2: // http://www.boost.org/LICENSE_1_0.txt)
williamr@2: //=======================================================================
williamr@2: 
williamr@2: #ifndef BOOST_GRAPH_TEST_HPP
williamr@2: #define BOOST_GRAPH_TEST_HPP
williamr@2: 
williamr@2: #include <vector>
williamr@2: #include <boost/test/minimal.hpp>
williamr@2: #include <boost/graph/filtered_graph.hpp>
williamr@2: #include <boost/graph/iteration_macros.hpp>
williamr@2: #include <boost/graph/isomorphism.hpp>
williamr@2: #include <boost/graph/copy.hpp>
williamr@2: #include <boost/graph/graph_utility.hpp> // for connects
williamr@2: 
williamr@2: 
williamr@2: // UNDER CONSTRUCTION 
williamr@2: 
williamr@2: namespace boost {
williamr@2: 
williamr@2:   template <typename Graph>
williamr@2:   struct graph_test
williamr@2:   {
williamr@2:   
williamr@2:     typedef typename graph_traits<Graph>::vertex_descriptor vertex_t;
williamr@2:     typedef typename graph_traits<Graph>::edge_descriptor edge_t;
williamr@2:     typedef typename graph_traits<Graph>::vertices_size_type v_size_t;
williamr@2:     typedef typename graph_traits<Graph>::degree_size_type deg_size_t;
williamr@2:     typedef typename graph_traits<Graph>::edges_size_type e_size_t;
williamr@2:     typedef typename graph_traits<Graph>::out_edge_iterator out_edge_iter;
williamr@2:     typedef typename property_map<Graph, vertex_index_t>::type index_map_t;
williamr@2:     typedef iterator_property_map<typename std::vector<vertex_t>::iterator,
williamr@2:       index_map_t,vertex_t,vertex_t&> IsoMap;
williamr@2: 
williamr@2:     struct ignore_vertex {
williamr@2:       ignore_vertex() { }
williamr@2:       ignore_vertex(vertex_t v) : v(v) { }
williamr@2:       bool operator()(vertex_t x) const { return x != v; }
williamr@2:       vertex_t v;
williamr@2:     };
williamr@2:     struct ignore_edge {
williamr@2:       ignore_edge() { }
williamr@2:       ignore_edge(edge_t e) : e(e) { }
williamr@2:       bool operator()(edge_t x) const { return x != e; }
williamr@2:       edge_t e;
williamr@2:     };
williamr@2:     struct ignore_edges {
williamr@2:       ignore_edges(vertex_t s, vertex_t t, const Graph& g) 
williamr@2:         : s(s), t(t), g(g) { }
williamr@2:       bool operator()(edge_t x) const { 
williamr@2:         return !(source(x, g) == s && target(x, g) == t);
williamr@2:       }
williamr@2:       vertex_t s; vertex_t t; const Graph& g;
williamr@2:     };
williamr@2: 
williamr@2:     //=========================================================================
williamr@2:     // Traversal Operations
williamr@2: 
williamr@2:     void test_incidence_graph
williamr@2:        (const std::vector<vertex_t>& vertex_set,
williamr@2:         const std::vector< std::pair<vertex_t, vertex_t> >& edge_set,
williamr@2:         const Graph& g)
williamr@2:     {
williamr@2:       typedef typename std::vector<vertex_t>::const_iterator vertex_iter;
williamr@2:       typedef typename std::vector< std::pair<vertex_t, vertex_t> >
williamr@2:         ::const_iterator edge_iter;
williamr@2:       typedef typename graph_traits<Graph>::out_edge_iterator out_edge_iter;
williamr@2: 
williamr@2:       for (vertex_iter ui = vertex_set.begin(); ui != vertex_set.end(); ++ui) {
williamr@2:         vertex_t u = *ui;
williamr@2:         std::vector<vertex_t> adj;
williamr@2:         for (edge_iter e = edge_set.begin(); e != edge_set.end(); ++e)
williamr@2:           if (e->first == u)
williamr@2:             adj.push_back(e->second);
williamr@2:         
williamr@2:         std::pair<out_edge_iter, out_edge_iter> p = out_edges(u, g);
williamr@2:         BOOST_CHECK(out_degree(u, g) == adj.size());
williamr@2:         BOOST_CHECK(deg_size_t(std::distance(p.first, p.second))
williamr@2:                    == out_degree(u, g));
williamr@2:         for (; p.first != p.second; ++p.first) {
williamr@2:           edge_t e = *p.first;
williamr@2:           BOOST_CHECK(source(e, g) == u);
williamr@2:           BOOST_CHECK(contains(adj, target(e, g)) == true);
williamr@2:         }
williamr@2:       }
williamr@2:     }
williamr@2: 
williamr@2:     void test_bidirectional_graph
williamr@2:       (const std::vector<vertex_t>& vertex_set,
williamr@2:        const std::vector< std::pair<vertex_t, vertex_t> >& edge_set,
williamr@2:        const Graph& g)
williamr@2:     {
williamr@2:       typedef typename std::vector<vertex_t>::const_iterator vertex_iter;
williamr@2:       typedef typename std::vector< std::pair<vertex_t, vertex_t> >
williamr@2:         ::const_iterator edge_iter;
williamr@2:       typedef typename graph_traits<Graph>::in_edge_iterator in_edge_iter;
williamr@2: 
williamr@2:       for (vertex_iter vi = vertex_set.begin(); vi != vertex_set.end(); ++vi) {
williamr@2:         vertex_t v = *vi;
williamr@2:         std::vector<vertex_t> inv_adj;
williamr@2:         for (edge_iter e = edge_set.begin(); e != edge_set.end(); ++e)
williamr@2:           if (e->second == v)
williamr@2:             inv_adj.push_back(e->first);
williamr@2: 
williamr@2:         std::pair<in_edge_iter, in_edge_iter> p = in_edges(v, g);
williamr@2:         BOOST_CHECK(in_degree(v, g) == inv_adj.size());
williamr@2:         BOOST_CHECK(deg_size_t(std::distance(p.first, p.second))
williamr@2:                    == in_degree(v, g));
williamr@2:         for (; p.first != p.second; ++p.first) {
williamr@2:           edge_t e = *p.first;
williamr@2:           BOOST_CHECK(target(e, g) == v);
williamr@2:           BOOST_CHECK(contains(inv_adj, source(e, g)) == true);
williamr@2:         }
williamr@2:       }
williamr@2:     }
williamr@2: 
williamr@2:     void test_adjacency_graph
williamr@2:       (const std::vector<vertex_t>& vertex_set,
williamr@2:        const std::vector< std::pair<vertex_t,vertex_t> >& edge_set,
williamr@2:        const Graph& g)
williamr@2:     {
williamr@2:       typedef typename std::vector<vertex_t>::const_iterator vertex_iter;
williamr@2:       typedef typename std::vector<std::pair<vertex_t,vertex_t> >
williamr@2:         ::const_iterator edge_iter;
williamr@2:       typedef typename graph_traits<Graph>::adjacency_iterator adj_iter;
williamr@2: 
williamr@2:       for (vertex_iter ui = vertex_set.begin(); ui != vertex_set.end(); ++ui) {
williamr@2:         vertex_t u = *ui;
williamr@2:         std::vector<vertex_t> adj;
williamr@2:         for (edge_iter e = edge_set.begin(); e != edge_set.end(); ++e)
williamr@2:           if (e->first == u)
williamr@2:             adj.push_back(e->second);
williamr@2: 
williamr@2:         std::pair<adj_iter, adj_iter> p = adjacent_vertices(u, g);
williamr@2:         BOOST_CHECK(deg_size_t(std::distance(p.first, p.second)) == adj.size());
williamr@2:         for (; p.first != p.second; ++p.first) {
williamr@2:           vertex_t v = *p.first;
williamr@2:           BOOST_CHECK(contains(adj, v) == true);
williamr@2:         }
williamr@2:       }
williamr@2:     }      
williamr@2: 
williamr@2:     void test_vertex_list_graph
williamr@2:       (const std::vector<vertex_t>& vertex_set, const Graph& g)
williamr@2:     {
williamr@2:       typedef typename graph_traits<Graph>::vertex_iterator v_iter;
williamr@2:       std::pair<v_iter, v_iter> p = vertices(g);
williamr@2:       BOOST_CHECK(num_vertices(g) == vertex_set.size());
williamr@2:       v_size_t n = std::distance(p.first, p.second);
williamr@2:       BOOST_CHECK(n == num_vertices(g));
williamr@2:       for (; p.first != p.second; ++p.first) {
williamr@2:         vertex_t v = *p.first;
williamr@2:         BOOST_CHECK(contains(vertex_set, v) == true);
williamr@2:       }
williamr@2:     }
williamr@2: 
williamr@2:     void test_edge_list_graph
williamr@2:       (const std::vector<vertex_t>& vertex_set, 
williamr@2:        const std::vector< std::pair<vertex_t, vertex_t> >& edge_set, 
williamr@2:        const Graph& g)
williamr@2:     {
williamr@2:       typedef typename graph_traits<Graph>::edge_iterator e_iter;
williamr@2:       std::pair<e_iter, e_iter> p = edges(g);
williamr@2:       BOOST_CHECK(num_edges(g) == edge_set.size());
williamr@2:       e_size_t m = std::distance(p.first, p.second);
williamr@2:       BOOST_CHECK(m == num_edges(g));
williamr@2:       for (; p.first != p.second; ++p.first) {
williamr@2:         edge_t e = *p.first;
williamr@2:         BOOST_CHECK(any_if(edge_set, connects(source(e, g), target(e, g), g)));
williamr@2:         BOOST_CHECK(contains(vertex_set, source(e, g)) == true);
williamr@2:         BOOST_CHECK(contains(vertex_set, target(e, g)) == true);
williamr@2:       }
williamr@2:     }
williamr@2: 
williamr@2:     void test_adjacency_matrix
williamr@2:       (const std::vector<vertex_t>& vertex_set, 
williamr@2:        const std::vector< std::pair<vertex_t, vertex_t> >& edge_set, 
williamr@2:        const Graph& g)
williamr@2:     {
williamr@2:       std::pair<edge_t, bool> p;
williamr@2:       for (typename std::vector<std::pair<vertex_t, vertex_t> >
williamr@2:              ::const_iterator i = edge_set.begin();
williamr@2:            i != edge_set.end(); ++i) {
williamr@2:         p = edge(i->first, i->second, g);
williamr@2:         BOOST_CHECK(p.second == true);
williamr@2:         BOOST_CHECK(source(p.first, g) == i->first);
williamr@2:         BOOST_CHECK(target(p.first, g) == i->second);
williamr@2:       }
williamr@2:       typename std::vector<vertex_t>::const_iterator j, k;
williamr@2:       for (j = vertex_set.begin(); j != vertex_set.end(); ++j)
williamr@2:         for (k = vertex_set.begin(); k != vertex_set.end(); ++k) {
williamr@2:           p = edge(*j, *k, g);
williamr@2:           if (p.second == true)
williamr@2:             BOOST_CHECK(any_if(edge_set, 
williamr@2:               connects(source(p.first, g), target(p.first, g), g)) == true);
williamr@2:         }
williamr@2:     }
williamr@2: 
williamr@2:     //=========================================================================
williamr@2:     // Mutating Operations
williamr@2: 
williamr@2:     void test_add_vertex(Graph& g)
williamr@2:     {
williamr@2:       Graph cpy;
williamr@2:       std::vector<vertex_t> iso_vec(num_vertices(g));
williamr@2:       IsoMap iso_map(iso_vec.begin(), get(vertex_index, g));
williamr@2:       copy_graph(g, cpy, orig_to_copy(iso_map));
williamr@2: 
williamr@2:       assert((verify_isomorphism(g, cpy, iso_map)));
williamr@2: 
williamr@2:       vertex_t v = add_vertex(g);
williamr@2:       
williamr@2:       BOOST_CHECK(num_vertices(g) == num_vertices(cpy) + 1);
williamr@2: 
williamr@2:       BOOST_CHECK(out_degree(v, g) == 0);
williamr@2: 
williamr@2:       // Make sure the rest of the graph stayed the same
williamr@2:       BOOST_CHECK((verify_isomorphism
williamr@2:                   (make_filtered_graph(g, keep_all(), ignore_vertex(v)), cpy,
williamr@2:                    iso_map)));
williamr@2:     }
williamr@2:     
williamr@2:     void test_add_edge(vertex_t u, vertex_t v, Graph& g)
williamr@2:     {
williamr@2:       Graph cpy;
williamr@2:       std::vector<vertex_t> iso_vec(num_vertices(g));
williamr@2:       IsoMap iso_map(iso_vec.begin(), get(vertex_index, g));
williamr@2:       copy_graph(g, cpy, orig_to_copy(iso_map));
williamr@2: 
williamr@2:       bool parallel_edge_exists = contains(adjacent_vertices(u, g), v);
williamr@2:       
williamr@2:       std::pair<edge_t, bool> p = add_edge(u, v, g);
williamr@2:       edge_t e = p.first;
williamr@2:       bool added = p.second;
williamr@2: 
williamr@2:       if (is_undirected(g) && u == v) // self edge
williamr@2:         BOOST_CHECK(added == false);
williamr@2:       else if (parallel_edge_exists)
williamr@2:         BOOST_CHECK(allows_parallel_edges(g) && added == true
williamr@2:                    || !allows_parallel_edges(g) && added == false);
williamr@2:       else
williamr@2:         BOOST_CHECK(added == true);
williamr@2: 
williamr@2:       if (p.second == true) { // edge added
williamr@2:         BOOST_CHECK(num_edges(g) == num_edges(cpy) + 1);
williamr@2:         
williamr@2:         BOOST_CHECK(contains(out_edges(u, g), e) == true);
williamr@2:         
williamr@2:         BOOST_CHECK((verify_isomorphism
williamr@2:                     (make_filtered_graph(g, ignore_edge(e)), cpy, iso_map)));
williamr@2:       }
williamr@2:       else { // edge not added
williamr@2:         if (! (is_undirected(g) && u == v)) {
williamr@2:           // e should be a parallel edge
williamr@2:           BOOST_CHECK(source(e, g) == u);
williamr@2:           BOOST_CHECK(target(e, g) == v);
williamr@2:         }
williamr@2:         // The graph should not be changed.
williamr@2:         BOOST_CHECK((verify_isomorphism(g, cpy, iso_map)));
williamr@2:       }
williamr@2:     } // test_add_edge()
williamr@2: 
williamr@2: 
williamr@2:     void test_remove_edge(vertex_t u, vertex_t v, Graph& g)
williamr@2:     {
williamr@2:       Graph cpy;
williamr@2:       std::vector<vertex_t> iso_vec(num_vertices(g));
williamr@2:       IsoMap iso_map(iso_vec.begin(), get(vertex_index, g));
williamr@2:       copy_graph(g, cpy, orig_to_copy(iso_map));
williamr@2: 
williamr@2:       deg_size_t occurances = count(adjacent_vertices(u, g), v);
williamr@2: 
williamr@2:       remove_edge(u, v, g);
williamr@2:       
williamr@2:       BOOST_CHECK(num_edges(g) + occurances == num_edges(cpy));
williamr@2:       BOOST_CHECK((verify_isomorphism
williamr@2:                   (g, make_filtered_graph(cpy, ignore_edges(u,v,cpy)),
williamr@2:                    iso_map)));
williamr@2:     }
williamr@2: 
williamr@2:     void test_remove_edge(edge_t e, Graph& g)
williamr@2:     {
williamr@2:       Graph cpy;
williamr@2:       std::vector<vertex_t> iso_vec(num_vertices(g));
williamr@2:       IsoMap iso_map(iso_vec.begin(), get(vertex_index, g));
williamr@2:       copy_graph(g, cpy, orig_to_copy(iso_map));
williamr@2: 
williamr@2:       vertex_t u = source(e, g), v = target(e, g);
williamr@2:       deg_size_t occurances = count(adjacent_vertices(u, g), v);
williamr@2:       
williamr@2:       remove_edge(e, g);
williamr@2: 
williamr@2:       BOOST_CHECK(num_edges(g) + 1 == num_edges(cpy));
williamr@2:       BOOST_CHECK(count(adjacent_vertices(u, g), v) + 1 == occurances);
williamr@2:       BOOST_CHECK((verify_isomorphism
williamr@2:                   (g, make_filtered_graph(cpy, ignore_edge(e)),
williamr@2:                    iso_map)));
williamr@2:     }
williamr@2: 
williamr@2:     void test_clear_vertex(vertex_t v, Graph& g)
williamr@2:     {
williamr@2:       Graph cpy;
williamr@2:       std::vector<vertex_t> iso_vec(num_vertices(g));
williamr@2:       IsoMap iso_map(iso_vec.begin(), get(vertex_index, g));
williamr@2:       copy_graph(g, cpy, orig_to_copy(iso_map));
williamr@2: 
williamr@2:       clear_vertex(v, g);
williamr@2: 
williamr@2:       BOOST_CHECK(out_degree(v, g) == 0);
williamr@2:       BOOST_CHECK(num_vertices(g) == num_vertices(cpy));
williamr@2:       BOOST_CHECK((verify_isomorphism
williamr@2:                   (g, make_filtered_graph(cpy, keep_all(), ignore_vertex(v)),
williamr@2:                    iso_map)));
williamr@2:     }
williamr@2: 
williamr@2:     //=========================================================================
williamr@2:     // Property Map
williamr@2: 
williamr@2:     template <typename PropVal, typename PropertyTag>
williamr@2:     void test_readable_vertex_property_graph
williamr@2:       (const std::vector<PropVal>& vertex_prop, PropertyTag, const Graph& g)
williamr@2:     {
williamr@2:       typedef typename property_map<Graph, PropertyTag>::const_type const_Map;
williamr@2:       const_Map pmap = get(PropertyTag(), g);
williamr@2:       typename std::vector<PropVal>::const_iterator i = vertex_prop.begin();
williamr@2: 
williamr@2:   for (typename boost::graph_traits<Graph>::vertex_iterator 
williamr@2:            bgl_first_9 = vertices(g).first, bgl_last_9 = vertices(g).second;
williamr@2:        bgl_first_9 != bgl_last_9; bgl_first_9 = bgl_last_9)
williamr@2:     for (typename boost::graph_traits<Graph>::vertex_descriptor v;
williamr@2:          bgl_first_9 != bgl_last_9 ? (v = *bgl_first_9, true) : false;
williamr@2:          ++bgl_first_9) {
williamr@2:       //BGL_FORALL_VERTICES_T(v, g, Graph) {
williamr@2:         typename property_traits<const_Map>::value_type 
williamr@2:           pval1 = get(pmap, v), pval2 = get(PropertyTag(), g, v);
williamr@2:         BOOST_CHECK(pval1 == pval2);
williamr@2:         BOOST_CHECK(pval1 == *i++);
williamr@2:       }
williamr@2:     }
williamr@2: 
williamr@2:     template <typename PropVal, typename PropertyTag>
williamr@2:     void test_vertex_property_graph
williamr@2:       (const std::vector<PropVal>& vertex_prop, PropertyTag tag, Graph& g)
williamr@2:     {
williamr@2:       typedef typename property_map<Graph, PropertyTag>::type PMap;
williamr@2:       PMap pmap = get(PropertyTag(), g);
williamr@2:       typename std::vector<PropVal>::const_iterator i = vertex_prop.begin();
williamr@2:   for (typename boost::graph_traits<Graph>::vertex_iterator 
williamr@2:            bgl_first_9 = vertices(g).first, bgl_last_9 = vertices(g).second;
williamr@2:        bgl_first_9 != bgl_last_9; bgl_first_9 = bgl_last_9)
williamr@2:     for (typename boost::graph_traits<Graph>::vertex_descriptor v;
williamr@2:          bgl_first_9 != bgl_last_9 ? (v = *bgl_first_9, true) : false;
williamr@2:          ++bgl_first_9)
williamr@2:       //      BGL_FORALL_VERTICES_T(v, g, Graph)
williamr@2:         put(pmap, v, *i++);
williamr@2: 
williamr@2:       test_readable_vertex_property_graph(vertex_prop, tag, g);
williamr@2: 
williamr@2:       BGL_FORALL_VERTICES_T(v, g, Graph)
williamr@2:         put(pmap, v, vertex_prop[0]);
williamr@2:       
williamr@2:       typename std::vector<PropVal>::const_iterator j = vertex_prop.begin();
williamr@2:       BGL_FORALL_VERTICES_T(v, g, Graph)
williamr@2:         put(PropertyTag(), g, v, *j++);
williamr@2:       
williamr@2:       test_readable_vertex_property_graph(vertex_prop, tag, g);      
williamr@2:     }
williamr@2:     
williamr@2:     
williamr@2:   };
williamr@2: 
williamr@2: 
williamr@2: } // namespace boost
williamr@2: 
williamr@2: #include <boost/graph/iteration_macros_undef.hpp>
williamr@2: 
williamr@2: #endif // BOOST_GRAPH_TEST_HPP