def test_use_style_attributes_default(style_key, barbell_network):
for node, node_attr in barbell_network.nodes.data():
node_attr[style_key] = 'TEST'
node_styles = generate_node_styles(barbell_network,
use_attributes())
for node, style in node_styles.items():
assert node in barbell_network
assert style[style_key] == 'TEST'
def test_generate_node_styles_dict(barbell_network):
updated = {'color': 'red'}
node_styles = generate_node_styles(barbell_network,
updated)
expected = default_node_style()
expected['color'] = 'red'
for node, style in node_styles.items():
assert node in barbell_network
assert style == expected
def test_use_style_attributes_filter_single(barbell_network):
for node, node_attr in barbell_network.nodes.data():
for style_key in _ALL_STYLE_KEYS:
node_attr[style_key] = 'TEST'
node_styles = generate_node_styles(barbell_network,
use_attributes('color'))
expected = default_node_style()
expected['color'] = 'TEST'
for node, style in node_styles.items():
assert node in barbell_network
assert style == expected
def test_generate_node_styles_callable(barbell_network):
def style_func(attrs):
return {'color': 'red'}
node_styles = generate_node_styles(barbell_network,
style_func)
expected = default_node_style()
expected['color'] = 'red'
for node, style in node_styles.items():
assert node in barbell_network
assert style == expected
def test_generate_node_style_typerror(barbell_network):
with pytest.raises(TypeError):
node_styles = generate_node_styles(barbell_network,
'red')