def testUpdateState(self):
matches = {}
u = Vertex('u1')
v = Vertex('v1')
g = Graph()
g._updateState(u, v, matches)
self.assertEquals(len(g._matchHistory), 1)
self.assertEquals(len(matches), 1)
self.assertEquals(matches[u.id], 'v1')
def testRestoreState(self):
# Save state with a couple calls to updateState() and then see that
# they are undone.
g = Graph()
matches = {}
u1 = Vertex('u1')
v1 = Vertex('v1')
g._updateState(u1, v1, matches) # u1 matched with v1
u2 = Vertex('u2')
v2 = Vertex('v2')
g._updateState(u2, v2, matches) # u2 matched with v2
# Now we should get the dictionary with only the original set of
# matches.
matches = g._restoreState(matches)
self.assertEquals(len(matches), 1)
self.assertTrue('u1' in matches)
self.assertTrue(matches['u1'], 'v1')
# Now we should have an empty dictionary.
matches = g._restoreState(matches)
self.assertEquals(len(matches), 0)