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) )