def test_constructor_and__str__(self): ne = NetworkEdge() self.assertEqual(None, ne.get_attributes()) self.assertEqual(None, ne.get_id()) self.assertEqual(None, ne.get_source_node_id()) self.assertEqual(None, ne.get_target_node_id()) self.assertEqual(None, ne.get_interaction()) self.assertEqual(None, ne.get_source_node_name()) self.assertEqual(None, ne.get_target_node_name()) self.assertEqual('po=None, s=None (None), t=None (None), i=None', str(ne)) # try with values set ne = NetworkEdge(edge_id=1, source_node_id=2, source_node_name='src', target_node_id=3, target_node_name='target', interaction='interaction', attributes=[]) self.assertEqual([], ne.get_attributes()) self.assertEqual(1, ne.get_id()) self.assertEqual(2, ne.get_source_node_id()) self.assertEqual(3, ne.get_target_node_id()) self.assertEqual('interaction', ne.get_interaction()) self.assertEqual('src', ne.get_source_node_name()) self.assertEqual('target', ne.get_target_node_name()) self.assertEqual('po=1, s=2 (src), t=3 (target), i=interaction', str(ne))
def test_getters_and_setters(self): ne = NetworkEdge() ne.set_attributes([]) ne.set_id(1) ne.set_source_node_id(2) ne.set_target_node_id(3) ne.set_source_node_name('src') ne.set_target_node_name('target') ne.set_interaction('interaction') self.assertEqual([], ne.get_attributes()) self.assertEqual(1, ne.get_id()) self.assertEqual(2, ne.get_source_node_id()) self.assertEqual(3, ne.get_target_node_id()) self.assertEqual('interaction', ne.get_interaction()) self.assertEqual('src', ne.get_source_node_name()) self.assertEqual('target', ne.get_target_node_name())