Function test

recipes/Python/576912_A_Tree_Finder/recipe-576912.py:141–150

(test_data)

  import random
  import time
  def test(test_data):
    fd = StringIO.StringIO()
    fd.write(test_data)
    fd.seek(0)
    r = Reader(fd)
    for case_id, edges in enumerate(r.gen_edges()):
      tree_finder = TreeFinder()
      tree_finder.add_edges(edges)
      s = '' if tree_finder.isTree() else 'not '
      print 'Case %d is %sa tree' % (case_id+1, s)

  simple_test_data = '''8 1
7 3 6 2 8 9 7 5