def test_staticentity_constructor(seven_by_six):
sbs = seven_by_six
ent = StaticEntity(arr=sbs.arr, labels=sbs.labels)
assert ent.size() == 6
assert len(ent.uidset) == 6
assert len(ent.children) == 7
assert isinstance(ent.incidence_dict["I"], list)
assert "I" in ent
assert "K" in ent
def test_staticentity_property(harry_potter):
arr = harry_potter.arr
labels = harry_potter.labels
ent = StaticEntity(arr=arr, labels=labels)
assert len(ent.keys) == 5
assert len(ent.uidset) == 7
assert len(ent.elements) == 7
assert isinstance(ent.elements['Hufflepuff'], list)
assert ent.is_empty(2) == False
assert len(ent.incidence_dict['Gryffindor']) == 6
assert ent.keyindex('House') == 0
def test_staticentity_attributes(harry_potter):
arr = harry_potter.arr
labels = harry_potter.labels
ent = StaticEntity(arr=arr, labels=labels)
assert isinstance(ent.arr, np.ndarray)
assert isinstance(ent.data, np.ndarray)
assert ent.data.shape == ent.dataframe.shape
assert isinstance(ent.labels, dict)
assert ent.dimensions == (7, 11, 10, 36, 26)
assert ent.dimsize == 5
assert len(ent.labs(0)) == 7
df = ent.dataframe
assert list(df.columns) == [
'House', 'Blood status', 'Species', 'Hair colour', 'Eye colour'
]
assert ent.dimensions == tuple(df.nunique())
assert list(ent.labels['House']) == list(df['House'].unique())
def test_staticentity_elements_by_level(seven_by_six):
sbs = seven_by_six
ent = StaticEntity(arr=sbs.arr, labels=sbs.labels)
assert ent.elements_by_level(0)
def test_staticentity_uidset_by_level(seven_by_six):
sbs = seven_by_six
ent = StaticEntity(arr=sbs.arr, labels=sbs.labels)
ent.uidset_by_level(0) == {'I', 'L', 'O', 'P', 'R', 'S'}
ent.uidset_by_level(1) == {'A', 'C', 'E', 'K', 'T1', 'T2', 'V'}
def test_staticentity_level(seven_by_six):
sbs = seven_by_six
ent = StaticEntity(arr=sbs.arr, labels=sbs.labels)
assert ent.level("I") == (0, 5)
assert ent.level("K") == (1, 3)
assert ent.level("K", max_level=0) == None
def test_staticentity_custom_attributes(harry_potter):
arr = harry_potter.arr
labels = harry_potter.labels
ent = StaticEntity(arr=arr, labels=labels)
assert ent.__len__() == 7
assert isinstance(ent.__str__(), str)
assert isinstance(ent.__repr__(), str)
assert isinstance(ent.__contains__('Muggle'), bool)
assert ent.__contains__('Muggle') == True
assert ent.__getitem__('Slytherin') == [
'Half-blood', 'Pure-blood', 'Pure-blood or half-blood'
]
assert isinstance(ent.__iter__(), Iterable)
assert isinstance(ent.__call__(), Iterable)
assert ent.__call__().__next__() == 'Unknown House'
def test_static_hypergraph_get_name(seven_by_six):
sbs = seven_by_six
H = Hypergraph(StaticEntity(arr=sbs.arr, labels=sbs.labels))
assert H.get_name(1) == 'C'
assert H.get_name(1, edges=True) == 'R'
def test_staticentity_construct_from_entity(seven_by_six):
sbs = seven_by_six
ent = StaticEntity(entity=sbs.edgedict)
assert len(ent.elements) == 6
def test_restrict_to_indices(harry_potter):
arr = harry_potter.arr
labels = harry_potter.labels
ent = StaticEntity(arr=arr, labels=labels)
assert ent.restrict_to_indices([1,
2]).uidset == {'Gryffindor', 'Ravenclaw'}
def test_restrict_to_levels(harry_potter):
arr = harry_potter.arr
labels = harry_potter.labels
ent = StaticEntity(arr=arr, labels=labels)
assert len(ent.restrict_to_levels([0]).uidset) == 7
def test_staticentity_turn_entity_data_into_dataframe(seven_by_six):
sbs = seven_by_six
ent = StaticEntity(arr=sbs.arr, labels=sbs.labels)
subset = ent.data[0:5]
assert ent.turn_entity_data_into_dataframe(subset).shape == (5, 2)
def test_staticentity_index(seven_by_six):
sbs = seven_by_six
ent = StaticEntity(arr=sbs.arr, labels=sbs.labels)
assert ent.index("nodes") == 1
assert ent.index("nodes", "K") == (1, 3)
def test_staticentity_translate_arr(seven_by_six):
sbs = seven_by_six
ent = StaticEntity(arr=sbs.arr, labels=sbs.labels)
assert ent.translate_arr((0, 0)) == ["P", "A"]
def test_staticentity_translate(seven_by_six):
sbs = seven_by_six
ent = StaticEntity(arr=sbs.arr, labels=sbs.labels)
assert ent.translate(0, 0) == "P"
assert ent.translate(1, [3, 4]) == ["K", "T1"]
def test_staticentity_incidence_matrix(seven_by_six):
sbs = seven_by_six
ent = StaticEntity(arr=sbs.arr, labels=sbs.labels)
assert ent.incidence_matrix(1, 0).todense().shape == (6, 7)
def test_staticentity_indices(seven_by_six):
sbs = seven_by_six
ent = StaticEntity(arr=sbs.arr, labels=sbs.labels)
assert ent.indices("nodes", "K") == [3]
assert ent.indices("nodes", ["K", "T1"]) == [3, 4]
def test_static_hypergraph_get_id(seven_by_six):
sbs = seven_by_six
H = Hypergraph(StaticEntity(arr=sbs.arr, labels=sbs.labels))
assert H.get_id('V') == 6
assert H.get_id('S', edges=True) == 2