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