From 5c72a9a310c3c3d6eb73de4ed80d6e97f12030b5 Mon Sep 17 00:00:00 2001 From: Ajda Pretnar Date: Fri, 14 Feb 2020 15:29:54 +0100 Subject: [PATCH] Add tests --- .../utils/tests/test_owbasevectorizer.py | 27 +++++++++++++++++++ 1 file changed, 27 insertions(+) diff --git a/orangecontrib/text/widgets/utils/tests/test_owbasevectorizer.py b/orangecontrib/text/widgets/utils/tests/test_owbasevectorizer.py index e69de29bb..47248eeeb 100644 --- a/orangecontrib/text/widgets/utils/tests/test_owbasevectorizer.py +++ b/orangecontrib/text/widgets/utils/tests/test_owbasevectorizer.py @@ -0,0 +1,27 @@ +from orangecontrib.text import Corpus +from orangecontrib.text.vectorization import BowVectorizer +from orangecontrib.text.widgets.utils.owbasevectorizer import OWBaseVectorizer +from orangewidget.tests.base import WidgetTest + + +class TestableBaseVectWidget(OWBaseVectorizer): + name = "TBV" + Method = BowVectorizer + + +class TestOWBaseVectorizer(WidgetTest): + def setUp(self): + self.widget = self.create_widget(TestableBaseVectWidget) + self.corpus = Corpus.from_file('deerwester') + + def test_hide_attributes(self): + self.send_signal("Corpus", self.corpus) + self.assertTrue(all(f.attributes['hidden'] for f in + self.get_output("Corpus").domain.attributes)) + self.widget.controls.hidden_cb.setChecked(False) + self.assertFalse(all(f.attributes['hidden'] for f in + self.get_output("Corpus").domain.attributes)) + new_corpus = Corpus.from_file('book-excerpts')[:10] + self.send_signal("Corpus", new_corpus) + self.assertFalse(all(f.attributes['hidden'] for f in + self.get_output("Corpus").domain.attributes))