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')