def test_entity_set_from_dict(seven_by_six):
sbs = seven_by_six
eset = EntitySet('sbs',sbs.edgedict)
M, rowdict, coldict = eset.incidence_matrix(index = True)
assert len(rowdict) == 7
assert len(coldict) == 6
x = np.ones(6).transpose()
assert np.max(M.dot(x)) == 3
def test_entity_set():
eset = EntitySet('eset1', {'A', 'C', 'K'})
eset2 = eset.clone('eset2')
assert eset.elements == eset2.elements
assert len(eset) == 3
assert eset.uid == 'eset1'
assert len(eset.children) == 0
with pytest.raises(HyperNetXError) as excinfo:
eset2.add(Entity('Z',['A','C','B']))
assert 'Fails the bipartite condition' in str(excinfo.value)
def test_entity_set():
eset = EntitySet("eset1", {"A", "C", "K"})
eset2 = eset.clone("eset2")
assert eset.elements == eset2.elements
assert len(eset) == 3
assert eset.uid == "eset1"
assert len(eset.children) == 0
with pytest.raises(HyperNetXError) as excinfo:
eset2.add(Entity("Z", ["A", "C", "B"]))
assert "Fails the bipartite condition" in str(excinfo.value)
def test_hypergraph_from_entity_set(seven_by_six):
sbs = seven_by_six
entityset = EntitySet("_", sbs.edgedict)
H = Hypergraph(entityset)
assert H.edges.incidence_dict == sbs.edgedict
assert H.s_degree("A") == 3
assert H.dim('O') == 1
assert len(H.edge_size_dist()) == 6
assert len(H.edge_neighbors("S")) == 4
def test_hypergraph_from_entity_set(seven_by_six):
sbs = seven_by_six
entityset = EntitySet('_', sbs.edgedict)
H = Hypergraph(entityset)
assert H.edges.incidence_dict == sbs.edgedict