def test_KMapperWrapper(self):
mapper = KMapperWrapper(verbose=0)
data = np.random.rand(100, 3)
graph = mapper.fit_map(data)
assert mapper.lens_.shape[0] == data.shape[0]
assert graph.get('nodes') and graph.get('links')
return
def test_init(self):
mapper = KMapperWrapper(verbose=0, memory=None)
data = np.random.rand(100, 3)
graph = mapper.fit_map(data)
dG = DyNeuGraph(G=graph)
assert dG.A.shape[0] == len(graph['nodes'])
assert dG.TCM.shape[0] == data.shape[0]
return
def test_init_with_y(self):
mapper = KMapperWrapper(verbose=0)
data = np.random.rand(100, 3)
graph = mapper.fit_map(data)
y = mapper.lens_
dG = DyNeuGraph(G=graph, y=y)
assert dG.A.shape[0] == len(graph['nodes'])
assert dG.TCM.shape[0] == data.shape[0]
return
def test_visualize_show(self):
mapper = KMapperWrapper(verbose=0)
data = np.random.rand(100, 3)
graph = mapper.fit_map(data)
y = mapper.lens_
dG = DyNeuGraph(G=graph, y=y)
# visualize results in a tempdir
with tempfile.TemporaryDirectory() as temp_dir:
temp_html = os.path.join(temp_dir, 'test.html')
dG.visualize(temp_html, path_assets=temp_dir, show=True)
assert os.path.exists(temp_html)
return
def test_density_filter(self):
mapper = KMapperWrapper(verbose=0)
data_a = np.random.rand(100, 3)
data_b = np.random.rand(100, 3) * 10
lens_a = density_filter(data_a, k=3)
lens_b = density_filter(data_b, k=3)
assert data_a.shape[0] == lens_a.shape[0]
assert data_b.shape[0] == lens_b.shape[0]
assert lens_a.shape[1] == 3
assert lens_b.shape[1] == 3
# make sure that b has lower lowest density
assert lens_a.min() > lens_b.min()
return