def test_flags(self):
model = TimeSeriesModelFixedResolution(
TimeSeriesFixedResolution("2019-07-05T12:00", "2 hours", [-5.0, 7.0], True, False)
)
model_index = model.index(0, 0)
self.assertEqual(model.flags(model_index), Qt.ItemIsSelectable | Qt.ItemIsEnabled)
model_index = model.index(0, 1)
self.assertEqual(model.flags(model_index), Qt.ItemIsSelectable | Qt.ItemIsEnabled | Qt.ItemIsEditable)
def test_batch_set_data(self):
model = TimeSeriesModelFixedResolution(
TimeSeriesFixedResolution("2019-07-05T12:00", "2 hours", [2.3, -5.0, 7.0], True, False)
)
indexes = [model.index(0, 0), model.index(1, 1), model.index(2, 1)]
values = ["1999-01-01T12:00", 55.5, -55.5]
model.batch_set_data(indexes, values)
expected = TimeSeriesFixedResolution("2019-07-05T12:00", "2 hours", [2.3, 55.5, -55.5], True, False)
self.assertEqual(model.value, expected)
def test_data(self):
model = TimeSeriesModelFixedResolution(
TimeSeriesFixedResolution("2019-07-05T12:00", "2 hours", [-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_setData(self):
model = TimeSeriesModelFixedResolution(
TimeSeriesFixedResolution("2019-07-05T12:00", "2 hours", [2.3, -5.0], True, False)
)
model_index = model.index(0, 1)
model.setData(model_index, -4.0)
self.assertEqual(
model.value, TimeSeriesFixedResolution("2019-07-05T12:00", "2 hours", [-4.0, -5.0], True, False)
)