def test_finds_a_link(self):
nerve = GraphNerve()
groups = {"a": [1, 2, 3, 4], "b": [1, 2, 3, 4]}
links, _ = nerve.compute(groups)
assert "a" in links or "b" in links
assert links["a"] == ["b"] or links["b"] == ["a"]
def test_pass_through_result(self):
nerve = GraphNerve()
groups = {"a": [1], "b": [2]}
res = dict()
links, _ = nerve.compute(groups)
assert res == links
def test_simplices(self):
nerve = GraphNerve()
groups = {"a": [1, 2, 3, 4], "b": [1, 2, 5], "c": [1, 5, 6, 7]}
links, simplices = nerve.compute(groups)
simplices = [set(s) for s in simplices]
assert {"a", "b"} in simplices
assert {"a", "b", "c"} not in simplices
assert max([len(s) for s in simplices]) == 2
assert len([n for n in simplices if len(n) == 1]) == 3
assert len([n for n in simplices if len(n) == 2]) == 3
def test_graphnerve(self):
nerve = GraphNerve()
groups = {"a": [1, 2, 3, 4], "b": [1, 2, 5], "c": [5, 6, 7]}
links, simplices = nerve.compute(groups)
# all vertices are simplices
assert all([[k] in simplices for k in groups])
# graph should be a -- b -- c
assert "c" not in links["a"] and "a" not in links["c"]
assert "b" in links["a"] or "a" in links["b"]
assert "c" in links["b"] or "b" in links["c"]
def test_min_intersection(self):
nerve = GraphNerve(min_intersection=2)
groups = {"a": [1, 2, 3, 4], "b": [1, 2, 5], "c": [5, 6, 7]}
links, simplices = nerve(groups)
# all vertices are simplices
assert all([[k] in simplices for k in groups])
# graph should be a -- b c
assert "c" not in links["a"] and "a" not in links["c"]
assert "b" in links["a"] or "a" in links["b"]
assert "c" not in links["b"] and "b" not in links["c"]
def test_no_link(self):
nerve = GraphNerve()
groups = {"a": [1, 2, 3, 4], "b": [5, 6, 7]}
links, _ = nerve.compute(groups)
assert not links