File size: 352 Bytes
19b102a |
1 2 3 4 5 6 7 8 9 10 11 |
import copy
import pytest
@pytest.mark.parametrize('model', [('kmeans_pca_topic_model'),
('base_topic_model'),
('custom_topic_model')])
def test_term_rank(model, request):
topic_model = copy.deepcopy(request.getfixturevalue(model))
fig = topic_model.visualize_term_rank()
|