self.model.rowCount(None)) def test_data_returned(self): indexes = [self.model.createIndex(x, 0) for x in range(3)] results = [self.model.data(index, Qt.DisplayRole) for index in indexes] self.assertEquals(self.testing_database_list, results) class gui_client_test(unittest.TestCase): """A TestCase for the methods in the gui_client class from the gui module""" def setUp(self): self.gui = gui_client(mpd_mock) def test_getting_playlists(self): playlists = self.gui.playlists.playlists self.assertEquals(len(playlists), self.gui.playlists.rowCount()) self.assertTrue('Goodies' in playlists) self.assertTrue('New_Python_Proba' in playlists) def test_toggle_text_change_on_different_events(self): self.gui.actionStop.activate(QAction.Trigger) self.assertEquals('Play', self.gui.actionToggle.text()) self.gui.actionToggle.activate(QAction.Trigger) self.assertEquals('Pause', self.gui.actionToggle.text()) self.gui.actionToggle.activate(QAction.Trigger) self.assertEquals('Play', self.gui.actionToggle.text()) if __name__ == '__main__': app = QApplication(sys.argv) unittest.main() app._exec()