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}})