def test_vector(self):
layer = Mock()
layer.subsetString.return_value = ""
layer.minimumValue.return_value = "1970-01-01 00:01:00"
layer.maximumValue.return_value = "1970-01-01 00:04:20"
layer.uniqueValues.return_value = ["1970-01-01 00:04:20", "1970-01-01 00:01:00"]
settings = ls.LayerSettings()
settings.layer = layer
settings.startTimeAttribute = self.from_attr
settings.endTimeAttribute = self.to_attr
vector = TimeVectorLayer(settings, iface=Mock())
assert (vector.getTimeFormat() == DEFAULT_FORMAT)
td = timedelta(minutes=5)
vector.setTimeRestriction(datetime(1970, 1, 1, 0, 0, 2), td)
layer.setSubsetString.assert_called_with(STRING_FORMAT.format(self.from_attr,
self.comparison_op,
"1970-01-01 00:05:02",
self.to_attr, "1970-01-01 "
"00:00:02"))
vector.setTimeRestriction(datetime(1980, 1, 1, 0, 0, 2), td)
layer.setSubsetString.assert_called_with(STRING_FORMAT.format(self.from_attr,
self.comparison_op,
"1980-01-01 00:05:02",
self.to_attr,
"1980-01-01 "
"00:00:02"))
def test_vector(self):
layer = Mock()
layer.subsetString.return_value = ""
layer.minimumValue.return_value = "1970-01-01 00:01:00"
layer.maximumValue.return_value = "1970-01-01 00:04:20"
layer.uniqueValues.return_value = [
"1970-01-01 00:04:20", "1970-01-01 00:01:00"
]
settings = ls.LayerSettings()
settings.layer = layer
settings.startTimeAttribute = self.from_attr
settings.endTimeAttribute = self.to_attr
vector = TimeVectorLayer(settings, iface=Mock())
assert (vector.getTimeFormat() == DEFAULT_FORMAT)
td = timedelta(minutes=5)
vector.setTimeRestriction(datetime(1970, 1, 1, 0, 0, 2), td)
layer.setSubsetString.assert_called_with(
STRING_FORMAT.format(self.from_attr, self.comparison_op,
"1970-01-01 00:05:02", self.to_attr,
"1970-01-01 "
"00:00:02"))
vector.setTimeRestriction(datetime(1980, 1, 1, 0, 0, 2), td)
layer.setSubsetString.assert_called_with(
STRING_FORMAT.format(self.from_attr, self.comparison_op,
"1980-01-01 00:05:02", self.to_attr,
"1980-01-01 "
"00:00:02"))
def test_vector(self):
layer = Mock()
provider = Mock()
layer.dataProvider.return_value = provider
layer.subsetString.return_value = ""
provider.minimumValue.return_value = "1970-01-01 00:01:00"
provider.maximumValue.return_value = "1970-01-01 00:04:20"
provider.storageType.return_value = 'PostgreSQL database with PostGIS extension'
vector = TimeVectorLayer(layer,
fromTimeAttribute=self.from_attr,
toTimeAttribute=self.to_attr,
enabled=True,
timeFormat=DEFAULT_FORMAT,
offset=0)
assert (vector.getTimeFormat() == DEFAULT_FORMAT)
td = timedelta(minutes=5)
vector.setTimeRestriction(datetime(1970, 1, 1, 0, 0, 2), td)
layer.setSubsetString.assert_called_with(
STRING_FORMAT.format(self.from_attr, self.comparison_op,
"1970-01-01 00:05:02", self.to_attr,
"1970-01-01 "
"00:00:02"))
vector.setTimeRestriction(datetime(1980, 1, 1, 0, 0, 2), td)
layer.setSubsetString.assert_called_with(
STRING_FORMAT.format(self.from_attr, self.comparison_op,
"1980-01-01 00:05:02", self.to_attr,
"1980-01-01 "
"00:00:02"))
def _test_spatialite_layer(self, attr, layer, is_int=False):
timeLayer = timevectorlayer.TimeVectorLayer(layer, attr, attr, True,
time_util.DEFAULT_FORMAT,
0)
self.tlm.registerTimeLayer(timeLayer)
if is_int:
self.assertEquals(timeLayer.getDateType(),
timevectorlayer.DateTypes.IntegerTimestamps)
self.assertEquals(timeLayer.getTimeFormat(), time_util.UTC)
else:
self.assertEquals(timeLayer.getDateType(),
timevectorlayer.DateTypes.DatesAsStrings)
self.assertEquals(timeLayer.getTimeFormat(),
time_util.DEFAULT_FORMAT)
start_time = time_util.str_to_datetime(timeLayer.getMinMaxValues()[0],
time_util.DEFAULT_FORMAT)
self.assertEquals(time_util.epoch_to_datetime(STARTTIME), start_time)
self.tlm.setTimeFrameType("minutes")
self.tlm.stepForward()
assert (start_time +
timedelta(minutes=1) == self.tlm.getCurrentTimePosition())
# only one feature is selected now, because there is one feature per minute
self.assertEquals(layer.featureCount(), 1)
FS = 5
self.tlm.setTimeFrameSize(FS)
# we have one feature per minute
self.assertEquals(layer.featureCount(), FS)
subsetString = layer.subsetString()
if is_int:
expectedSubsetString = INT_FORMAT.format(
attr, self.comparison_op,
time_util.datetime_to_epoch(self.tlm.getCurrentTimePosition() +
timedelta(minutes=FS)), attr,
time_util.datetime_to_epoch(self.tlm.getCurrentTimePosition()))
self.assertEquals(subsetString, expectedSubsetString)
minimum_bound_seconds = int(subsetString.split(" ")[6])
self.assertEquals(
self.tlm.getCurrentTimePosition(),
time_util.epoch_to_datetime(minimum_bound_seconds))
if not is_int:
self.assertEqual(timeLayer.getTimeFormat(),
time_util.DEFAULT_FORMAT)
expectedSubsetString = STRING_FORMAT.format(
attr, self.comparison_op,
time_util.datetime_to_str(
self.tlm.getCurrentTimePosition() + timedelta(minutes=FS),
timeLayer.getTimeFormat()), attr,
time_util.datetime_to_str(self.tlm.getCurrentTimePosition(),
timeLayer.getTimeFormat()))
self.assertEquals(subsetString, expectedSubsetString)
self.tlm.stepForward()
self.assertEquals(layer.featureCount(), FS)
def _test_spatialite_layer(self, attr, layer, is_int=False):
settings = LayerSettings()
settings.layer = layer
settings.startTimeAttribute = attr
settings.endTimeAttribute = attr
timeLayer = timevectorlayer.TimeVectorLayer(settings, iface=Mock())
self.tlm.registerTimeLayer(timeLayer)
if is_int:
self.assertEquals(timeLayer.getDateType(), timevectorlayer.DateTypes.IntegerTimestamps)
self.assertEquals(timeLayer.getTimeFormat(), time_util.UTC)
else:
self.assertEquals(timeLayer.getDateType(), timevectorlayer.DateTypes.DatesAsStrings)
self.assertEquals(timeLayer.getTimeFormat(), time_util.DEFAULT_FORMAT)
start_time = time_util.str_to_datetime(timeLayer.getMinMaxValues()[0], time_util.PENDING)
self.assertEquals(time_util.epoch_to_datetime(STARTTIME), start_time)
self.tlm.setTimeFrameType("minutes")
self.tlm.stepForward()
assert start_time + timedelta(minutes=1) == self.tlm.getCurrentTimePosition()
# only one feature is selected now, because there is one feature per minute
self.assertEquals(layer.featureCount(), 1)
FS = 5
self.tlm.setTimeFrameSize(FS)
# we have one feature per minute
self.assertEquals(layer.featureCount(), FS)
subsetString = layer.subsetString()
if is_int:
expectedSubsetString = INT_FORMAT.format(
attr,
self.comparison_op,
time_util.datetime_to_epoch(self.tlm.getCurrentTimePosition() + timedelta(minutes=FS)),
attr,
time_util.datetime_to_epoch(self.tlm.getCurrentTimePosition()),
)
self.assertEquals(subsetString, expectedSubsetString)
minimum_bound_seconds = int(subsetString.split(" ")[6])
self.assertEquals(self.tlm.getCurrentTimePosition(), time_util.epoch_to_datetime(minimum_bound_seconds))
if not is_int:
self.assertEqual(timeLayer.getTimeFormat(), time_util.DEFAULT_FORMAT)
expectedSubsetString = STRING_FORMAT.format(
attr,
self.comparison_op,
time_util.datetime_to_str(
self.tlm.getCurrentTimePosition() + timedelta(minutes=FS), timeLayer.getTimeFormat()
),
attr,
time_util.datetime_to_str(self.tlm.getCurrentTimePosition(), timeLayer.getTimeFormat()),
)
self.assertEquals(subsetString, expectedSubsetString)
self.tlm.stepForward()
self.assertEquals(layer.featureCount(), FS)
