def test_get_id(self): v_1 = Vertex('A') v_2 = Vertex('B') v_3 = Vertex('C') self.assertEqual(v_1.get_id(), 'A') self.assertNotEqual(v_2.get_id(), 'A') self.assertEqual(v_3.get_id(), 'C')
def test_init(self): vertex_1 = Vertex('A') assert vertex_1.get_id() is 'A' assert vertex_1.neighbors == {}