def test_add_node(self): ig = InfluenceGraph() ig.add_node(1, activation=12) ig.add_node(2, activation=13) ig.add_node(3, activation=0) self.assertEqual(ig.nodes(), { 1: { 'activation': 12 }, 2: { 'activation': 13 }, 3: { 'activation': 0 } })
def test_remove_node(self): ig = InfluenceGraph() ig.add_node(1, activation=12) ig.add_node(2, activation=13) ig.remove_node(2) self.assertEqual(ig.nodes(), {1: {'activation': 12}})