def test_init_edge_colors():
"""
Does two checks:
1. If edge_color is passed in as a keyword argument, check that
self.edge_colors is a list with more than one element.
2. If edge_color is not passed in as a keyword argument, check that
self.edge_colors is a list of 'black', of length 0.
"""
G = make_graph_for_edges() # noqa
b = BasePlot(graph=G, edge_color="weight")
assert len(set(b.edge_colors)) > 1
G = make_graph_for_grouping() # noqa
b = BasePlot(graph=G)
assert len(set(b.edge_colors)) == 0
def test_init_node_colors():
"""
Does two checks:
1. If node_color is not passed in as a keyword argument, check that
self.node_colors is a list of 'blue', of length (number of nodes).
2. If node_color is passed in as a keyword argument, check that
self.node_colors is a list with more than one element.
"""
G = make_graph_for_grouping() # noqa
b = BasePlot(graph=G, node_color="year")
assert len(set(b.node_colors)) > 1
G = make_graph_for_grouping() # noqa
b = BasePlot(graph=G)
assert len(set(b.node_colors)) == 1
def test_initialization():
"""Tests initialization of plot object."""
n_nodes = 10
G = nx.erdos_renyi_graph(n=n_nodes, p=0.3) # noqa
b = BasePlot(graph=G)
assert len(b.nodes) == len(G.nodes())
def test_init_font():
"""
Checks if the passed value for font and fontfamily
is set
"""
G = make_graph_for_edges()
fontfamily = "fantasy"
fontsize = 5
b = BasePlot(graph=G, fontsize=fontsize, fontfamily=fontfamily)
assert b.fontfamily == fontfamily
assert b.fontsize == fontsize
def test_init_data_types():
"""
Checks that the data_types dictionary is initialized correctly.
"""
G = make_graph_for_grouping() # noqa
b = BasePlot(
graph=G, data_types={"year": "ordinal", "affiliation": "categorical"}
)
assert isinstance(b.data_types, dict)
def test_init_data_types():
"""
Checks that the data_types dictionary is initialized correctly.
"""
G = make_graph_for_grouping()
b = BasePlot(graph=G,
data_types={
'year': 'ordinal',
'affiliation': 'categorical'
})
assert isinstance(b.data_types, dict)
def test_init_sort_nodes():
"""
Tests initialization with sorting of nodes.
This tests that the nodes are ordered correctly when sorted on the
"node_order" key.
"""
G = make_graph_for_grouping() # noqa
b = BasePlot(graph=G, node_order="year")
assert b.nodes == [
n for n, d in sorted(G.nodes(data=True), key=lambda x: x[1]["year"])
]
def test_init_group_nodes():
"""
Tests initialization with grouping of nodes.
This only tests that the nodes are ordered correctly when sorted on the
`node_grouping` key.
"""
G = make_graph_for_grouping() # noqa
b = BasePlot(graph=G, node_grouping="affiliation")
assert b.nodes == [
n for n, d in sorted(G.nodes(data=True),
key=lambda x: x[1]["affiliation"])
]
def test_init_sort_and_group_nodes():
"""
Tests initialization with sorting and grouping of nodes.
This tests that the nodes are ordered correctly when first grouped on the
`node_grouping` key, and then sorted within each group on the `node_order`
key.
"""
G = make_graph_for_grouping() # noqa
b = BasePlot(graph=G, node_grouping='affiliation', node_order='year')
assert b.nodes == [
n for n, d in sorted(G.nodes(data=True),
key=lambda x: (x[1]['affiliation'], x[1]['year']))
]
