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

Function test_dynamic

tests/test_variations/test_dynamic.py:15–22  ·  view source on GitHub ↗
(model, documents, request)

Source from the content-addressed store, hash-verified

13 ],
14)
15def test_dynamic(model, documents, request):
16 topic_model = copy.deepcopy(request.getfixturevalue(model))
17 timestamps = [i % 10 for i in range(len(documents))]
18 topics_over_time = topic_model.topics_over_time(documents, timestamps)
19
20 assert topics_over_time.Frequency.sum() == len(documents)
21 assert set(topics_over_time.Topic.unique()) == set(topic_model.topics_)
22 assert len(topics_over_time.Timestamp.unique()) == len(set(timestamps))

Callers

nothing calls this directly

Calls 1

topics_over_timeMethod · 0.80

Tested by

no test coverage detected