def test_create(self):
timeseries_group = mommy.make(models.TimeseriesGroup)
timeseries = models.Timeseries(type=models.Timeseries.AGGREGATED,
timeseries_group=timeseries_group)
timeseries.save()
self.assertEqual(models.Timeseries.objects.first().type,
models.Timeseries.AGGREGATED)
def test_many_aggregated_per_group(self):
timeseries_group = mommy.make(models.TimeseriesGroup, name="Temperature")
self._make_timeseries(timeseries_group, models.Timeseries.AGGREGATED)
models.Timeseries(
timeseries_group=timeseries_group,
type=models.Timeseries.AGGREGATED,
time_step="D",
).save()
def test_uniqueness(self):
timeseries_group = mommy.make(models.TimeseriesGroup, name="Temperature")
self._make_timeseries(timeseries_group, models.Timeseries.AGGREGATED)
with self.assertRaises(IntegrityError):
models.Timeseries(
timeseries_group=timeseries_group,
type=models.Timeseries.AGGREGATED,
time_step="H",
).save()
def test_only_one_regularized_per_group(self):
timeseries_group = mommy.make(models.TimeseriesGroup, name="Temperature")
self._make_timeseries(timeseries_group, models.Timeseries.REGULARIZED)
with self.assertRaises(IntegrityError):
models.Timeseries(
timeseries_group=timeseries_group,
type=models.Timeseries.REGULARIZED,
time_step="D",
).save()
def test_only_one_initial_per_group(self):
timeseries_group = mommy.make(models.TimeseriesGroup,
name="Temperature")
self._make_timeseries(timeseries_group, models.Timeseries.INITIAL)
with self.assertRaises(IntegrityError):
models.Timeseries(
timeseries_group=timeseries_group,
type=models.Timeseries.INITIAL,
time_step="D",
).save()
def test_create(self):
station = mommy.make(models.Station)
variable = mommy.make(models.Variable)
unit = mommy.make(models.UnitOfMeasurement)
time_zone = mommy.make(models.TimeZone)
timeseries = models.Timeseries(
gentity=station,
name="Temperature",
variable=variable,
unit_of_measurement=unit,
time_zone=time_zone,
precision=2,
)
timeseries.save()
self.assertEqual(models.Timeseries.objects.first().name, "Temperature")
def test_type_serialization_for_initial_when_nondefault_language(self):
serializer = TimeseriesSerializer(models.Timeseries(type=100))
with translation.override("el"):
self.assertEqual(serializer.data["type"], "Initial")
def test_type_serialization_for_checked(self):
serializer = TimeseriesSerializer(models.Timeseries(type=200))
self.assertEqual(serializer.data["type"], "Checked")
def test_type_serialization_for_initial(self):
serializer = TimeseriesSerializer(models.Timeseries(type=100))
self.assertEqual(serializer.data["type"], "Initial")
