def test_get_by_id_negative(self): n1, n2, n3, n4, n5, n6, n7, n8, n9, n10 = get_nodes() s = Nodes([n1, n2, n3]) with self.assertRaises(ValueError): s.from_id(4)
def test_get_by_id_positive(self): n1, n2, n3, n4, n5, n6, n7, n8, n9, n10 = get_nodes() s = Nodes([n1, n2, n3]) self.assertEqual(s.from_id(1), n1) self.assertEqual(s.from_id(2), n2) self.assertEqual(s.from_id(3), n3)