2 Created on Jul 28, 2010
8 from erange import makeGraphs
10 testFileName = "/tmp/testEdgeFileForUnitTests.txt"
12 class TestMakeGraphs(unittest.TestCase):
22 def testGetEdges(self):
24 self.assertEquals({}, makeGraphs.getEdges(nodeList))
26 nodeEntry = "ex_node1\tex_node2\t1"
27 nodeList.append(nodeEntry)
28 result = {"ex_node1": [("ex_node2", 1)],
29 "ex_node2": [("ex_node1", 1)]}
30 self.assertEquals(result, makeGraphs.getEdges(nodeList))
32 nodeEntry = "ex_node1\tex_node3\t2"
33 nodeList.append(nodeEntry)
34 result = {"ex_node1": [("ex_node2", 1), ("ex_node3", 2)],
35 "ex_node2": [("ex_node1", 1)],
36 "ex_node3": [("ex_node1", 2)]
38 self.assertEquals(result, makeGraphs.getEdges(nodeList))
40 result = {"node1": [("node2", 1), ("node3", 2)],
41 "node2": [("node1", 1)],
42 "node3": [("node1", 2)]
44 self.assertEquals(result, makeGraphs.getEdges(nodeList, shorten=True))
46 nodeEntry = "ex:node1\tex:node2\t1"
47 nodeList = [nodeEntry]
48 result = {"ex:node1": [("ex:node2", 1)],
49 "ex:node2": [("ex:node1", 1)]}
50 self.assertEquals(result, makeGraphs.getEdges(nodeList, shorten=True))
53 nodeList = [nodeEntry]
54 self.assertEquals({}, makeGraphs.getEdges(nodeList))
55 nodeEntry = "node1\tnode2\t1"
56 nodeList.append(nodeEntry)
57 result = {"node1": [("node2", 1)],
58 "node2": [("node1", 1)]}
59 self.assertEquals(result, makeGraphs.getEdges(nodeList))
62 def testGetEdgesFromFile(self):
63 self.edgeFile = open(testFileName, "w")
64 self.edgeFile.write("node1\tnode2\t1")
67 result = {"node1": [("node2", 1)],
68 "node2": [("node1", 1)]}
69 self.assertEquals(result, makeGraphs.getEdgesFromFile(testFileName))
71 os.remove(testFileName)
75 suite = unittest.TestSuite()
76 suite.addTest(unittest.makeSuite(TestMakeGraphs))
81 if __name__ == "__main__":
82 #import sys;sys.argv = ['', 'Test.testName']