1 parent 1c22736 commit 91c4e3cCopy full SHA for 91c4e3c
1 file changed
tests/test_reduction/test_delete.py
@@ -74,10 +74,12 @@ def test_delete_keeps_custom_labels(model, request):
74
75
remaining = [topic for topic in sorted(set(topic_model.topics_)) if topic != -1]
76
topic_to_delete = remaining[0]
77
+ before = len(set(topic_model.topics_))
78
79
topic_model.delete_topics([topic_to_delete])
80
81
unique_topics = sorted(set(topic_model.topics_))
82
assert isinstance(topic_model.custom_labels_, list)
83
assert len(topic_model.custom_labels_) == len(unique_topics)
- assert topic_to_delete not in unique_topics
84
+ # Remaining topics are remapped by frequency, so the deleted id can reappear.
85
+ assert len(unique_topics) < before
0 commit comments