#!/usr/bin/python

import unittest
import graph

class GraphTestCase(unittest.TestCase):
    def testEdge(self):
        g = graph.Graph(False)
        n1 = g.newNode()
        n2 = g.newNode()
        g.addEdge(n1, n2)
        self.assertTrue(g.hasEdge(n2, n1))
        self.assertTrue(g.hasEdge(n1, n2))
        g.delNode(n1)
        g.delNode(n2)

    def testDegree(self):
        g = graph.Graph(False)
        n1 = g.newNode()
        n2 = g.newNode()
        n3 = g.newNode()
        g.addEdge(n1, n2)
        g.addEdge(n1, n3)
        self.assertEqual(2, n1.Degree)
        self.assertEqual(1, n2.Degree)
        g.delNode(n2)
        self.assertEqual(1, n1.Degree)

class DigraphTestCase(unittest.TestCase):
    pass
        

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