def test_invalid_node_color_statistic_interactive():
pipe = make_mapper_pipeline()
graph = pipe.fit_transform(X_arr)
node_color_statistic = np.arange(len(graph.vs))
with pytest.raises(ValueError):
plot_interactive_mapper_graph(
pipe, X_arr, node_color_statistic=node_color_statistic
)
def test_pipeline_cloned(X, clone_pipeline, layout_dim,
color_by_columns_dropdown):
"""Verify that the pipeline is changed on interaction if and only if
`clone_pipeline` is False (with `layout_dim` set to 2 or 3)."""
# TODO: Monitor development of the ipytest project to convert these into
# true notebook tests integrated with pytest
params = {
"cover": {
"initial": {"n_intervals": 10, "kind": "uniform",
"overlap_frac": 0.1},
"new": {"n_intervals": 15, "kind": "balanced", "overlap_frac": 0.2}
},
"clusterer": {
"initial": {"affinity": "euclidean"},
"new": {"affinity": "manhattan"}
},
"contract_nodes": {"initial": True, "new": False},
"min_intersection": {"initial": 4, "new": 1},
}
pipe = make_mapper_pipeline(
cover=CubicalCover(**params["cover"]["initial"]),
clusterer=FirstSimpleGap(**params["clusterer"]["initial"]),
contract_nodes=params["contract_nodes"]["initial"],
min_intersection=params["min_intersection"]["initial"]
)
fig = plot_interactive_mapper_graph(
pipe, X, clone_pipeline=clone_pipeline, layout_dim=layout_dim,
color_by_columns_dropdown=color_by_columns_dropdown
)
# Get relevant widgets and change their states, then check final values
for step, values in params.items():
if step in ["cover", "clusterer"]:
for param_name, initial_param_value in values["initial"].items():
new_param_value = values["new"][param_name]
widgets = _get_widgets_by_trait(fig, "description", param_name)
for w in widgets:
w.set_state({'value': new_param_value})
final_param_value_actual = \
pipe.get_mapper_params()[f"{step}__{param_name}"]
final_param_value_expected = \
initial_param_value if clone_pipeline else new_param_value
assert final_param_value_actual == final_param_value_expected
else:
initial_param_value = values["initial"]
new_param_value = values["new"]
widgets = _get_widgets_by_trait(fig, "description", step)
for w in widgets:
w.set_state({'value': new_param_value})
final_param_value_actual = \
pipe.get_mapper_params()[f"{step}"]
final_param_value_expected = \
initial_param_value if clone_pipeline else new_param_value
assert final_param_value_actual == final_param_value_expected
def test_cluster_sizes(self):
"""Verify that the total number of calculated clusters is equal to
the number of displayed clusters."""
pipe = make_mapper_pipeline(clusterer=FirstSimpleGap())
warnings.simplefilter("ignore")
fig = plot_interactive_mapper_graph(pipe, X)
w_scatter = self._get_widget_by_trait(fig, 'data')
node_sizes_vis = [self._get_size_from_hovertext(s_)
for s_ in w_scatter.get_state()
['_data'][1]['hovertext']]
g = pipe.fit_transform(X)
node_size_real = [len(node)
for node in g['node_metadata']['node_elements']]
assert sum(node_sizes_vis) == sum(node_size_real)