test.py

利用C

"""Unit test for the graph library"""

__author__ = "Gustav Magnus Vikstrom ([e-mail])"
__date__ = "2007-02-12 -- 2007-03-21"
__copyright__ = "Copyright (C) 2007 Gustav Magnus Vikstrom"
__license__  = "GNU LGPL Version 2.1"

import unittest
import os
import tempfile
import numpy
from dolfin import *

class Editor(unittest.TestCase):

    def testUndirected(self):
        """Create undirected graph with edges added out of order (should pass)"""
        graph = Graph()
        editor = GraphEditor()
        editor.open(graph, "undirected")
        editor.initVertices(4)
        editor.addVertex(0, 2)
        editor.addVertex(1, 3)
        editor.addVertex(2, 2)
        editor.addVertex(3, 3)
        editor.initEdges(5)
        editor.addEdge(0, 1)
        editor.addEdge(1, 2)
        editor.addEdge(2, 3)
        editor.addEdge(1, 3)
        editor.addEdge(0, 3)
        editor.close()

        self.assertEqual(graph.numVertices(), 4)
        self.assertEqual(graph.numEdges(), 5)
        self.assertEqual(graph.numArches(), 10)

    def testDirected(self):
        """Create directed graph with edges added out of order (should pass)"""
        graph = Graph()
        editor = GraphEditor()
        editor.open(graph, "directed")
        editor.initVertices(4)
        editor.addVertex(0, 2)
        editor.addVertex(1, 1)
        editor.addVertex(2, 1)
        editor.addVertex(3, 1)
        editor.initEdges(5)
        editor.addEdge(0, 1)
        editor.addEdge(1, 2)
        editor.addEdge(2, 3)
        editor.addEdge(3, 1)
        editor.addEdge(0, 3)
        editor.close()

        self.assertEqual(graph.numVertices(), 4)
        self.assertEqual(graph.numEdges(), 5)
        self.assertEqual(graph.numArches(), 5)

class InputOutput(unittest.TestCase):

    def testUndirectedGraphXML(self):
        """Write and read undirected graph to/from file"""
        graph = UndirectedClique(4)

        # Create temp file from graph, read file and delete tempfile
        fd, tmp_name = tempfile.mkstemp(suffix=".xml")
        file = File(tmp_name)
        file << graph
        graph2 = Graph()
        file >> graph2
        os.remove(tmp_name)

        self.assertEqual(graph.numVertices(), graph2.numVertices())
        self.assertEqual(graph.numEdges(), graph2.numEdges())

    def testDirectedGraphXML(self):
        """Write and read directed graph to/from file"""
        graph = DirectedClique(4)

        # Create temp file from graph, read file and delete tempfile
        fd, tmp_name = tempfile.mkstemp(suffix=".xml")
        file = File(tmp_name)
        file << graph
        graph2 = Graph()
        file >> graph2
        os.remove(tmp_name)

        graph2.disp()
        self.assertEqual(graph.numVertices(), graph2.numVertices())
        self.assertEqual(graph.numEdges(), graph2.numEdges())

    def testMetisGraphConvertion(self):
        """Create metis graph file, convert to dolfin xml and read xml file"""

        fd, tmp_name = tempfile.mkstemp(suffix=".gra")
        mfile = open(tmp_name, "w")
        mfile.write("3 3\n")
        mfile.write(" 1 2\n")
        mfile.write(" 0 2\n")
        mfile.write(" 0 1\n")
        
        mfile.close()

        # Create temp dolfin xml file from metis graph
        fd, tmp_name = tempfile.mkstemp(suffix=".xml")
        os.system('dolfin-convert %s %s' % (mfile.name, tmp_name))

        # Read dolfin xml file
        graph = Graph()
        file = File(tmp_name)
        file >> graph

        # Delete tempfiles
        os.remove(tmp_name)
        os.remove(mfile.name)
        
        self.assertEqual(graph.numVertices(), 3)
        self.assertEqual(graph.numEdges(), 3)

    def testScotchGraphConvertion(self):
        """Create scotch graph file, convert to dolfin xml and read xml file"""
        fd, tmp_name = tempfile.mkstemp(suffix=".grf")
        sfile = open(tmp_name, "w")
        sfile.write("0\n")
        sfile.write("3 6\n")
        sfile.write("1 000\n")
        sfile.write("2 1 2\n")
        sfile.write("2 0 2\n")
        sfile.write("2 0 1\n")

        sfile.close()

        # Create temp dolfin xml file from scotch graph
        fd, tmp_name = tempfile.mkstemp(suffix=".xml")
        os.system('dolfin-convert %s %s' % (sfile.name, tmp_name))

        # Read dolfin xml file
        graph = Graph()
        file = File(tmp_name)
        file >> graph
        
        # Delete tempfiles
        os.remove(tmp_name)
        os.remove(sfile.name)

        self.assertEqual(graph.numVertices(), 3)
        self.assertEqual(graph.numEdges(), 3)

class Partitioning(unittest.TestCase):

    def testPartition(self):
        """Create a graph and partition it"""
  
        """ Example graph:
		             0 -- 1 -- 2
		                  |  /
				            | /
				       4 -- 3 -- 6
						 |    |    |
					    |		|    |
					    7		5 -- 8
        """

        nn = 9
        num_part = 2
        graph = Graph()
        editor = GraphEditor()
        editor.open(graph, "undirected")
        editor.initVertices(nn)
        editor.addVertex(0, 1)
        editor.addVertex(1, 3)
        editor.addVertex(2, 2)
        editor.addVertex(3, 5)
        editor.addVertex(4, 2)
        editor.addVertex(5, 2)
        editor.addVertex(6, 2)
        editor.addVertex(7, 1)
        editor.addVertex(8, 2)
        editor.initEdges(10)
        editor.addEdge(0, 1)
        editor.addEdge(1, 2)
        editor.addEdge(1, 3)
        editor.addEdge(2, 3)
        editor.addEdge(3, 4)
        editor.addEdge(3, 5)
        editor.addEdge(3, 6)
        editor.addEdge(4, 7)
        editor.addEdge(5, 8)
        editor.addEdge(6, 8)
        editor.close()

        parts = numpy.array(0, 'L')
        parts.resize(nn)

        GraphPartition.partition(graph, num_part, parts)
        GraphPartition.eval(graph, num_part, parts)
        GraphPartition.disp(graph, num_part, parts)
        edgecut = GraphPartition.edgecut(graph, num_part, parts)
        GraphPartition.check(graph, num_part, parts)

if __name__ == "__main__":
    unittest.main()