def test_membership(self):
mapper = KeplerMapper(verbose=0)
data = np.random.rand(100, 2)
graph = mapper.map(data)
n, m = list(graph["nodes"].items())[0]
g = to_networkx(graph)
assert g.nodes[n]["membership"] == m
def test_convertions(self):
mapper = KeplerMapper(verbose=0)
data = np.random.rand(100, 2)
graph = mapper.map(data)
g = to_networkx(graph)
assert isinstance(g, nx.Graph)
def test_nx_input(self, mapper):
draw_matplotlib(km.to_networkx(mapper))