def test_color_function_deprecated_replaced(): km = KeplerMapper() X, labels = make_circles(1000, random_state=0) lens = km.fit_transform(X, projection=[0]) color_values = lens[:, 0] sc = km.map(lens, X) X_names=[] lens_names=[] custom_tooltips = np.array(["customized_%s" % (l) for l in labels]) with warnings.catch_warnings(record=True) as w: # Cause all warnings to always be triggered. warnings.simplefilter("always") # TODO: plotlyviz.plotlyviz # plotlyviz.get_mapper_graph json_graph, mapper_summary, colorf_distribution = get_mapper_graph(sc, color_function=color_values) _test_raised_deprecation_warning(w) # plotlyviz.scomplex_to_graph _ = scomplex_to_graph( simplicial_complex=sc, color_function=color_values, X=X, X_names=X_names, lens=lens, lens_names=lens_names, custom_tooltips=custom_tooltips, colorscale=default_colorscale, ) _test_raised_deprecation_warning(w)
def test_color_function_deprecated_replaced(self, default_colorscale): mapper = KeplerMapper() data, labels = make_circles(1000, random_state=0) lens = mapper.fit_transform(data, projection=[0]) graph = mapper.map(lens, data) color_values = lens[:, 0] node_color_function = "mean" inverse_X = data projected_X = lens projected_X_names = [ "projected_%s" % (i) for i in range(projected_X.shape[1]) ] inverse_X_names = [ "inverse_%s" % (i) for i in range(inverse_X.shape[1]) ] custom_tooltips = np.array(["customized_%s" % (l) for l in labels]) # https://docs.python.org/3/library/warnings.html#testing-warnings with warnings.catch_warnings(record=True) as w: # Cause all warnings to always be triggered. warnings.simplefilter("always") # kmapper.visualize html = mapper.visualize(graph, color_function=lens, color_function_name="lens[:, 0]") _test_raised_deprecation_warning(w) # visuals._format_mapper_data graph_data = _format_mapper_data( graph=graph, color_function=color_values, node_color_function=node_color_function, X=inverse_X, X_names=inverse_X_names, lens=projected_X, lens_names=projected_X_names, custom_tooltips=custom_tooltips, ) _test_raised_deprecation_warning(w) # visuals._graph_data_distribution histogram = _graph_data_distribution( graph, color_function=lens, node_color_function=node_color_function, colorscale=default_colorscale, ) _test_raised_deprecation_warning(w)