From e29d040e2716ed76fbddbedd1fb113f688837cbe Mon Sep 17 00:00:00 2001 From: nojhan Date: Sun, 24 Jul 2011 08:04:29 +0200 Subject: [PATCH] test tsplib import/export on a simple square --- test_tsp.py | 35 +++++++++++++++++++++++++++++++++++ 1 file changed, 35 insertions(+) create mode 100644 test_tsp.py diff --git a/test_tsp.py b/test_tsp.py new file mode 100644 index 0000000..ae3e1b2 --- /dev/null +++ b/test_tsp.py @@ -0,0 +1,35 @@ + +from tsplib import * + + +segments = [ + ( (0,0),(0,2) ), + ( (0,2),(2,2) ), + ( (2,2),(2,0) ), + ( (2,0),(0,0) ) +] + +filename = "test.tsp" +with open(filename,"w") as fd: + write_segments( segments, fd=fd, size=1, depth=0, rounding=10 ) + write_segments( segments, fd=sys.stdout, size=1, depth=0, rounding=10 ) + +with open(filename,"r") as fd: + nodes = read_nodes( fd ) + +print "Nodes: id (x, y)" +for idx,node in nodes.items(): + print idx,node + +with open(filename,"r") as fd: + vertices = read_vertices( fd ) + +print "Segments: (x1,y1) (x2,y2)" +segments = [] +for i1,i2 in vertices: + print nodes[i1],nodes[i2] + segments.append( (nodes[i1],nodes[i2]) ) + +plot_segments( segments ) + +