RhodeCode

/* -*- C++ -*-
 *
 * This file is a part of LEMON, a generic C++ optimization library
 *
 * Copyright (C) 2003-2008
 * Egervary Jeno Kombinatorikus Optimalizalasi Kutatocsoport
 * (Egervary Research Group on Combinatorial Optimization, EGRES).
 *
 * Permission to use, modify and distribute this software is granted
 * provided that this copyright notice appears in all copies. For
 * precise terms see the accompanying LICENSE file.
 *
 * This software is provided "AS IS" with no warranty of any kind,
 * express or implied, and with no claim as to its suitability for any
 * purpose.
 *
 */

#ifndef LEMON_TEST_MAP_TEST_H
#define LEMON_TEST_MAP_TEST_H


#include <vector>
#include <lemon/maps.h>

#include "test_tools.h"


//! \ingroup misc
//! \file
//! \brief Some utilities to test map classes.

namespace lemon {



  template <typename Graph>
  void checkGraphNodeMap() {
    Graph graph;
    const int num = 16;
    
    typedef typename Graph::Node Node;

    std::vector<Node> nodes;
    for (int i = 0; i < num; ++i) {
      nodes.push_back(graph.addNode());      
    }
    typedef typename Graph::template NodeMap<int> IntNodeMap;
    IntNodeMap map(graph, 42);
    for (int i = 0; i < int(nodes.size()); ++i) {
      check(map[nodes[i]] == 42, "Wrong map constructor.");      
    }
    for (int i = 0; i < num; ++i) {
      nodes.push_back(graph.addNode());
      map[nodes.back()] = 23;
    }
    map = constMap<Node>(12);
    for (int i = 0; i < int(nodes.size()); ++i) {
      check(map[nodes[i]] == 12, "Wrong map constructor.");      
    }    
    graph.clear();
    nodes.clear();
  }

  template <typename Graph>
  void checkGraphArcMap() {
    Graph graph;
    const int num = 16;
    
    typedef typename Graph::Node Node;
    typedef typename Graph::Arc Arc;
    
    std::vector<Node> nodes;
    for (int i = 0; i < num; ++i) {
      nodes.push_back(graph.addNode());
    }
    
    std::vector<Arc> edges;
    for (int i = 0; i < num; ++i) {
      for (int j = 0; j < i; ++j) {
	edges.push_back(graph.addArc(nodes[i], nodes[j]));
      }
    }
    
    typedef typename Graph::template ArcMap<int> IntArcMap;
    IntArcMap map(graph, 42);
    
    for (int i = 0; i < int(edges.size()); ++i) {
      check(map[edges[i]] == 42, "Wrong map constructor.");      
    }
    
    for (int i = 0; i < num; ++i) {
      for (int j = i + 1; j < num; ++j) {
	edges.push_back(graph.addArc(nodes[i], nodes[j]));
	map[edges.back()] = 23;
      }
    }
    map = constMap<Arc>(12);
    for (int i = 0; i < int(edges.size()); ++i) {
      check(map[edges[i]] == 12, "Wrong map constructor.");      
    }    
    graph.clear();
    edges.clear();    
  }

  template <typename Graph>
  void checkGraphEdgeMap() {
    Graph graph;
    const int num = 16;
    
    typedef typename Graph::Node Node;
    typedef typename Graph::Edge Edge;
    
    std::vector<Node> nodes;
    for (int i = 0; i < num; ++i) {
      nodes.push_back(graph.addNode());
    }
    
    std::vector<Edge> edges;
    for (int i = 0; i < num; ++i) {
      for (int j = 0; j < i; ++j) {
	edges.push_back(graph.addEdge(nodes[i], nodes[j]));
      }
    }
    
    typedef typename Graph::template EdgeMap<int> IntEdgeMap;
    IntEdgeMap map(graph, 42);
    
    for (int i = 0; i < int(edges.size()); ++i) {
      check(map[edges[i]] == 42, "Wrong map constructor.");      
    }
    
    for (int i = 0; i < num; ++i) {
      for (int j = i + 1; j < num; ++j) {
	edges.push_back(graph.addEdge(nodes[i], nodes[j]));
	map[edges.back()] = 23;
      }
    }
    map = constMap<Edge>(12);
    for (int i = 0; i < int(edges.size()); ++i) {
      check(map[edges[i]] == 12, "Wrong map constructor.");      
    }    
    graph.clear();
    edges.clear();    
  }

}

#endif