def test_color_proper_rows(self):
"""Rows are colored corresponding to the document"""
model = ConcordanceModel()
model.set_width(2)
model.set_corpus(self.corpus)
model.set_word("of", DummyState)
color1 = model.data(model.index(0, 0), Qt.BackgroundRole)
self.assertEqual(model.data(model.index(1, 0), Qt.BackgroundRole),
color1)
self.assertNotEqual(model.data(model.index(2, 0), Qt.BackgroundRole),
color1)
def test_color_proper_rows(self):
"""Rows are colored corresponding to the document"""
model = ConcordanceModel()
model.set_width(2)
model.set_corpus(self.corpus)
model.set_word("of")
color1 = model.data(model.index(0, 0), Qt.BackgroundRole)
self.assertEqual(model.data(model.index(1, 0), Qt.BackgroundRole),
color1)
self.assertNotEqual(model.data(model.index(2, 0), Qt.BackgroundRole),
color1)
def test_data_non_displayroles(self):
"""Other possibly implemented roles return correct types"""
model = ConcordanceModel()
model.set_corpus(self.corpus)
model.set_word("of", DummyState)
ind00 = model.index(0, 0)
self.assertIsInstance(model.data(ind00, Qt.ForegroundRole),
(QBrush, type(None)))
self.assertIsInstance(model.data(ind00, Qt.BackgroundRole),
(QColor, type(None)))
self.assertIsInstance(model.data(ind00, Qt.TextAlignmentRole),
(Qt.Alignment, type(None)))
def test_data_non_displayroles(self):
"""Other possibly implemented roles return correct types"""
model = ConcordanceModel()
model.set_corpus(self.corpus)
model.set_word("of")
ind00 = model.index(0, 0)
self.assertIsInstance(model.data(ind00, Qt.ForegroundRole),
(QBrush, type(None)))
self.assertIsInstance(model.data(ind00, Qt.BackgroundRole),
(QColor, type(None)))
self.assertIsInstance(model.data(ind00, Qt.TextAlignmentRole),
(Qt.Alignment, type(None)))
def test_data(self):
"""data returns proper text to display"""
model = ConcordanceModel()
model.set_width(2)
self.assertEqual(model.rowCount(QModelIndex()), 0)
model.set_corpus(self.corpus)
model.set_word("of", DummyState)
# The same document in two rows
self.assertEqual(model.rowCount(QModelIndex()), 7)
self.assertEqual(model.data(model.index(0, 0)), "A survey")
self.assertEqual(model.data(model.index(0, 1)), "of")
self.assertEqual(model.data(model.index(0, 2)), "user opinion")
self.assertEqual(model.data(model.index(1, 0)), "user opinion")
self.assertEqual(model.data(model.index(1, 1)), "of")
self.assertEqual(model.data(model.index(1, 2)), "computer system")
# The third column has just a single word
self.assertEqual(model.data(model.index(2, 0)), "engineering testing")
self.assertEqual(model.data(model.index(2, 1)), "of")
self.assertEqual(model.data(model.index(2, 2)), "EPS")
# The first column has just a single word
self.assertEqual(model.data(model.index(3, 0)), "Relation")
self.assertEqual(model.data(model.index(3, 1)), "of")
self.assertEqual(model.data(model.index(3, 2)), "user perceived")
def test_data(self):
"""data returns proper text to display"""
model = ConcordanceModel()
model.set_width(2)
self.assertEqual(model.rowCount(QModelIndex()), 0)
model.set_corpus(self.corpus)
model.set_word("of")
# The same document in two rows
self.assertEqual(model.rowCount(QModelIndex()), 7)
self.assertEqual(model.data(model.index(0, 0)), "A survey")
self.assertEqual(model.data(model.index(0, 1)), "of")
self.assertEqual(model.data(model.index(0, 2)), "user opinion")
self.assertEqual(model.data(model.index(1, 0)), "user opinion")
self.assertEqual(model.data(model.index(1, 1)), "of")
self.assertEqual(model.data(model.index(1, 2)), "computer system")
# The third column has just a single word
self.assertEqual(model.data(model.index(2, 0)), "engineering testing")
self.assertEqual(model.data(model.index(2, 1)), "of")
self.assertEqual(model.data(model.index(2, 2)), "EPS")
# The first column has just a single word
self.assertEqual(model.data(model.index(3, 0)), "Relation")
self.assertEqual(model.data(model.index(3, 1)), "of")
self.assertEqual(model.data(model.index(3, 2)), "user perceived")
def test_set_word(self):
"""Concondance.set_word resets the indices"""
# Human machine interface for lab abc computer applications
model = ConcordanceModel()
model.set_corpus(self.corpus)
model.set_width(2)
model.set_word("of", DummyState)
self.assertEqual(model.rowCount(QModelIndex()), 7)
self.assertEqual(model.data(model.index(0, 0)), "A survey")
model.set_word("lab", DummyState)
self.assertEqual(model.rowCount(QModelIndex()), 1)
self.assertEqual(model.data(model.index(0, 0)), "interface for")
model.set_word(None, DummyState)
self.assertEqual(model.rowCount(QModelIndex()), 0)
def test_set_word(self):
"""Concondance.set_word resets the indices"""
# Human machine interface for lab abc computer applications
model = ConcordanceModel()
model.set_corpus(self.corpus)
model.set_width(2)
model.set_word("of")
self.assertEqual(model.rowCount(QModelIndex()), 7)
self.assertEqual(model.data(model.index(0, 0)), "A survey")
model.set_word("lab")
self.assertEqual(model.rowCount(QModelIndex()), 1)
self.assertEqual(model.data(model.index(0, 0)), "interface for")
model.set_word(None)
self.assertEqual(model.rowCount(QModelIndex()), 0)