MCPcopy
hub / github.com/MaartenGr/BERTopic / test_hierarchy

Function test_hierarchy

tests/test_variations/test_hierarchy.py:16–23  ·  view source on GitHub ↗
(model, documents, request)

Source from the content-addressed store, hash-verified

14 ],
15)
16def test_hierarchy(model, documents, request):
17 topic_model = copy.deepcopy(request.getfixturevalue(model))
18 hierarchical_topics = topic_model.hierarchical_topics(documents)
19
20 merged_topics = set([v for vals in hierarchical_topics.Topics.values for v in vals])
21
22 assert len(hierarchical_topics) > 0
23 assert merged_topics == set(topic_model.topics_).difference({-1})
24
25
26@pytest.mark.parametrize(

Callers

nothing calls this directly

Calls 1

hierarchical_topicsMethod · 0.80

Tested by

no test coverage detected