def test_cannot_remove_all_rows(self):
model = TimeSeriesModelVariableResolution(
TimeSeriesVariableResolution(
["2019-07-05T12:00", "2019-07-21T08:15", "2019-07-23T09:10"],
[2.3, -5.0, 7.0], True, False))
self.assertTrue(model.removeRows(0, 3))
self.assertEqual(
model.value,
TimeSeriesVariableResolution(["2019-07-05T12:00"], [2.3], True,
False))
def test_insertRows_at_the_beginning_with_only_one_value(self):
model = TimeSeriesModelVariableResolution(
TimeSeriesVariableResolution(["2019-07-05T12:00"], [-5.0], True,
False))
self.assertTrue(model.insertRows(0, 1))
self.assertEqual(
model.value,
TimeSeriesVariableResolution(
["2019-07-05T11:00", "2019-07-05T12:00"], [0.0, -5.0], True,
False),
)
def test_flags(self):
model = TimeSeriesModelVariableResolution(
TimeSeriesVariableResolution(
["2019-07-05T12:00", "2019-07-21T08:15"], [-5.0, 7.0], True,
False))
for row in range(2):
for column in range(2):
model_index = model.index(row, column)
self.assertEqual(
model.flags(model_index),
Qt.ItemIsSelectable | Qt.ItemIsEnabled | Qt.ItemIsEditable)
def test_removeRows_from_the_middle(self):
model = TimeSeriesModelVariableResolution(
TimeSeriesVariableResolution(
["2019-07-05T12:00", "2019-07-21T08:15", "2019-07-23T09:10"],
[2.3, -5.0, 7.0], True, False))
self.assertTrue(model.removeRows(1, 1))
self.assertEqual(
model.value,
TimeSeriesVariableResolution(
["2019-07-05T12:00", "2019-07-23T09:10"], [2.3, 7.0], True,
False))
def test_insertRows_in_the_end_with_only_one_value(self):
model = TimeSeriesModelVariableResolution(
TimeSeriesVariableResolution(["2019-07-05T12:00"], [-5.0], True,
False))
self.assertTrue(model.insertRows(1, 1))
self.assertEqual(
model.value,
TimeSeriesVariableResolution(
["2019-07-05T12:00", "2019-07-05T13:00"], [-5.0, 0.0], True,
False),
)
def test_insertRows_single_row_in_the_middle(self):
model = TimeSeriesModelVariableResolution(
TimeSeriesVariableResolution(
["2019-07-05T12:00", "2019-07-21T12:00"], [-5.0, 7.0], True,
False))
self.assertTrue(model.insertRows(1, 1))
self.assertEqual(
model.value,
TimeSeriesVariableResolution(
["2019-07-05T12:00", "2019-07-13T12:00", "2019-07-21T12:00"],
[-5.0, 0.0, 7.0], True, False),
)
def test_insertRows_in_the_end(self):
model = TimeSeriesModelVariableResolution(
TimeSeriesVariableResolution(
["2019-07-05T12:00", "2019-07-21T12:00"], [-5.0, 7.0], True,
False))
self.assertTrue(model.insertRows(2, 1))
self.assertEqual(
model.value,
TimeSeriesVariableResolution(
["2019-07-05T12:00", "2019-07-21T12:00", "2019-08-06T12:00"],
[-5.0, 7.0, 0.0], True, False),
)
def test_setData(self):
model = TimeSeriesModelVariableResolution(
TimeSeriesVariableResolution(
["1991-01-01T13:30", "1992-01-01T13:30"], [2.3, -5.0], True,
False))
model_index = model.index(0, 1)
model.setData(model_index, -4.0)
self.assertEqual(
model.value,
TimeSeriesVariableResolution(
["1991-01-01T13:30", "1992-01-01T13:30"], [-4.0, -5.0], True,
False),
)
def test_reset_updates_indexes(self):
model = TimeSeriesModelVariableResolution(
TimeSeriesVariableResolution(
["2019-07-05T12:00", "2019-07-21T08:15"], [2.3, -5.0], True,
False))
model.reset(
TimeSeriesVariableResolution(
["1991-01-01T13:30", "1992-01-01T13:30"], [7.0, -4.0], False,
True))
self.assertEqual(
model.value,
TimeSeriesVariableResolution(
["1991-01-01T13:30", "1992-01-01T13:30"], [7.0, -4.0], False,
True),
)
def test_batch_set_data(self):
model = TimeSeriesModelVariableResolution(
TimeSeriesVariableResolution(
["2019-07-05T12:00", "2019-07-21T08:15", "2019-07-23T09:10"],
[2.3, -5.0, 7.0], True, False))
indexes = [model.index(0, 0), model.index(1, 1), model.index(2, 1)]
values = ["2018-07-05T12:00", 55.5, -55.5]
model.batch_set_data(indexes, values)
expected = TimeSeriesVariableResolution(
["2018-07-05T12:00", "2019-07-21T08:15", "2019-07-23T09:10"],
[2.3, 55.5, -55.5], True, False)
self.assertEqual(model.value, expected)
def setUp(self):
QApplication.clipboard().clear()
self._table_view = IndexedValueTableView(parent=None)
series = TimeSeriesVariableResolution(
[
"2019-08-08T12:00", "2019-08-08T13:00", "2019-08-08T14:00",
"2019-08-08T15:00"
],
[1.1, 2.2, 3.3, 4.4],
False,
False,
)
model = TimeSeriesModelVariableResolution(series)
self._table_view.setModel(model)
def test_data(self):
model = TimeSeriesModelVariableResolution(
TimeSeriesVariableResolution(
["2019-07-05T12:00", "2019-07-21T08:15"], [-5.0, 7.0], True,
False))
for role in [Qt.DisplayRole, Qt.EditRole]:
model_index = model.index(0, 0)
self.assertEqual(model.data(model_index, role),
"2019-07-05T12:00:00")
model_index = model.index(0, 1)
self.assertEqual(model.data(model_index, role), -5.0)
def test_removing_last_row_fails(self):
model = TimeSeriesModelVariableResolution(
TimeSeriesVariableResolution(["2019-07-05T12:00"], [2.3], True,
False))
self.assertFalse(model.removeRows(0, 1))