digraphtools/test/test_topsort.py

   1 #! /usr/bin/env python
   2
   3 '''unit tests for digraphtools'''
   4
   5 import unittest2 as unittest
   6 import digraphtools
   7 import digraphtools.topsort as topsort
   8
   9 class TopsortTests(unittest.TestCase):
  10         def test_vr_topsort(self):
  11                 n = 5
  12                 partial_order = [(1,2), (2,3), (1,5)]
  13                 g = digraphtools.graph_from_edges(digraphtools.from_partial_order(partial_order))
  14                 grid = topsort.partial_order_to_grid(partial_order,n)
  15                 for le in topsort.vr_topsort(n,grid):
  16                         digraphtools.verify_partial_order(digraphtools.iter_partial_order(g), le)
  17
  18 if __name__ == '__main__':
  19         unittest.main()