class TestDeleteContract(unittest.TestCase):
def setUp(self):
self.uut = DeleteContract()
self.g = nx.Graph() # Triangle graph
self.g.add_edge(1, 2)
self.g.add_edge(2, 3)
self.g.add_edge(1, 3)
def test_getValue(self):
self.assertRaises(KeyError, self.uut.getValue, self.g)
self.uut.addGraph(self.g, 42)
self.assertEqual(42, self.uut.getValue(self.g))
# Check same graph but different node names gets valid result
g = nx.Graph([("a", "b"), ("b", "c"), ("a", "c")])
self.assertEqual(42, self.uut.getValue(g))
def test_addGraph(self):
self.uut.addGraph(self.g, 0)
self.assertEqual(1, len(self.uut.knownGraphs))
self.assertEqual([(self.g, 0)], self.uut.knownGraphs[(3, 3)])
def test_getValueEndPoints(self):
g = nx.Graph()
g.add_node(1)
self.assertEqual(1, self.uut.getValue(g))
def setUp(self):
self.uut = DeleteContract()
self.g = nx.Graph() # Triangle graph
self.g.add_edge(1, 2)
self.g.add_edge(2, 3)
self.g.add_edge(1, 3)
