def test_properties():
"""Check methods get_prop_value and set_prop_value."""
v = Vertex(1, {})
with pytest.raises(Exception):
assert v.get_prop_value("something") is None
v.set_prop_value("name", "value")
assert v.get_prop_value("name") == "value"
with pytest.raises(Exception):
# this statement still should raise an exception
assert v.get_prop_value("something") is None
def test_string_representation_4():
"""Check special method __str__ for vertices with properies."""
v = Vertex(1, {})
v.set_prop_value("name", "value")
s = str(v)
assert s == """
Vertex: 1
Neighbours: []
Properties: {'name': 'value'}
"""
v = Vertex(2, {})
v.set_prop_value("other", "XXX")
s = str(v)
assert s == """