diff --git a/orangecontrib/text/widgets/owconcordance.py b/orangecontrib/text/widgets/owconcordance.py index 5fc382ccc..d9c1ba25c 100644 --- a/orangecontrib/text/widgets/owconcordance.py +++ b/orangecontrib/text/widgets/owconcordance.py @@ -167,7 +167,7 @@ def get_data(self): data.append([" ".join(txt)]) docs.append([self.corpus.titles[self.word_index[row][0]]]) conc = np.array(np.hstack((data, docs)), dtype=object) - return Corpus(domain, metas=conc, text_features=[domain.metas[1]]) + return Corpus(domain, metas=conc, text_features=[domain.metas[0]]) class OWConcordance(OWWidget): diff --git a/orangecontrib/text/widgets/tests/test_owconcordances.py b/orangecontrib/text/widgets/tests/test_owconcordances.py index 2e1b15b27..dfaa99b34 100644 --- a/orangecontrib/text/widgets/tests/test_owconcordances.py +++ b/orangecontrib/text/widgets/tests/test_owconcordances.py @@ -141,7 +141,10 @@ def test_concordance_output(self): model = ConcordanceModel() model.set_word("of") model.set_corpus(self.corpus) - self.assertEqual(len(model.get_data()), 7) + output = model.get_data() + self.assertEqual(len(output), 7) + self.assertEqual(len(output.text_features), 1) + self.assertTrue(output.text_features[0].name.startswith('Conc. of')) class TestConcordanceWidget(WidgetTest):