def test_rowCount(self):
model = GAMSRecordListModel()
self.assertEqual(model.rowCount(), 0)
model.reset([["key1"]], "set")
self.assertEqual(model.rowCount(), 1)
def test_reset(self):
model = GAMSRecordListModel()
self.assertEqual(model.rowCount(), 0)
model.reset([("key1", "key2")], "set")
self.assertEqual(model.rowCount(), 1)
self.assertEqual(model.index(0, 0).data(), "key1, key2")
def test_headerData(self):
model = GAMSRecordListModel()
self.assertEqual(model.headerData(0, Qt.Horizontal), '')
self.assertEqual(model.headerData(0, Qt.Vertical), 1)
def test_moveRows_up(self):
model = GAMSRecordListModel()
model.reset([("key1",), ("key2",), ("key3",)], "set")
no_parent = QModelIndex()
self.assertTrue(model.moveRows(no_parent, 2, 1, no_parent, 1))
self.assertEqual(model.index(0, 0).data(), "key1")
self.assertEqual(model.index(1, 0).data(), "key3")
self.assertEqual(model.index(2, 0).data(), "key2")
self.assertTrue(model.moveRows(no_parent, 1, 1, no_parent, 0))
self.assertEqual(model.index(0, 0).data(), "key3")
self.assertEqual(model.index(1, 0).data(), "key1")
self.assertEqual(model.index(2, 0).data(), "key2")
self.assertFalse(model.moveRows(no_parent, 0, 1, no_parent, -1))
def test_data(self):
model = GAMSRecordListModel()
model.reset([("key1", "key2")], "set")
index = model.index(0, 0)
self.assertEqual(index.data(), "key1, key2")