def setUp(self):
self.conf = Configuration.load(TimeSerie.CONF_PATH, Ini)
self.timeserie = TimeSerie(self.conf)
points = [
(ts, 1) for ts in range(0, 24 * 3600, 3600)
]
self.timewindow = TimeWindow(start=points[0][0], stop=points[-1][0])
self.points = points
def test_intervals(self):
"""Test calculate on different intervals."""
now = time()
# let a period of 1 day
period = Period(day=1)
oneday = period.total_seconds()
rnow = period.round_timestamp(now)
# let a timewindow of 10+1/4 days
timewindow = TimeWindow(start=now - oneday, stop=now + 45/4 * oneday)
nan = float('nan')
points = [
# the first interval is empty
(rnow, nan), # the second interval contains nan at start
(rnow + oneday + 1, nan), # the third interval contains nan at start + 1
(rnow + 2 * oneday, 1), # the fourth interval contains 1 at start
(rnow + 3 * oneday + 1, 1), # the fourth interval contains 1 at start + 1
(rnow + 4 * oneday, nan), (rnow + 4 * oneday + 1, 1), # the fith interval contains 1 and nan
(rnow + 5 * oneday, 1), (rnow + 5 * oneday + 1, 1), # the sixth interval contains 1 and 1
(rnow + 6 * oneday, 1), (rnow + 6 * oneday, 1), # the sixth interval contains 1 and 1 at the same time
(rnow + 7 * oneday, nan), (rnow + 7 * oneday, nan), # the sixth interval contains nan and nan at the same time
]
timeserie = TimeSerie(
config=self.conf,
aggregation='sum',
period=period,
round_time=True
)
_points = timeserie.calculate(points, timewindow)
for i in [0, 1, 2, 5, 8, 9, 10, 11, 12]:
self.assertEqual(_points[i][0], rnow + (i - 1) * oneday)
self.assertTrue(isnan(_points[i][1]))
for i in [3, 4]:
self.assertEqual(_points[i][0], rnow + (i - 1) * oneday)
self.assertEqual(_points[i][1], 1)
for i in [6, 7]:
self.assertEqual(_points[i][0], rnow + (i - 1) * oneday)
self.assertEqual(_points[i][1], 2)
self.assertEqual(len(_points), len(points) + 1)
def test_scenario(self):
"""
Calculate aggregations over 5 years
"""
timewindow = self._five_years_timewidow()
# for all round_time values
for round_time in (True, False):
unit_length = 3600
# for all units
for index, unit in enumerate(Period.UNITS):
max_value_unit = Period.MAX_UNIT_VALUES[index]
if unit in (
Period.MICROSECOND,
Period.SECOND,
Period.MINUTE,
Period.WEEK,
Period.MONTH,
Period.YEAR
):
continue
value = randint(2, max_value_unit)
period = Period(**{unit: value})
kwargs = {'period': period}
period_length = unit_length * value
timeserie = TimeSerie(config=self.conf,
round_time=round_time,
**kwargs)
timesteps = timeserie.timesteps(timewindow)
timesteps_gap = timesteps[1] - timesteps[0]
self.assertEqual(timesteps_gap, period_length)
for i in range(5):
points = [
(t, random()) for t in
range(
int(timewindow.start()),
int(timewindow.stop()),
Period(**{unit: 1}).total_seconds()
)
]
aggregated_points = timeserie.calculate(points, timewindow)
len_aggregated_points = len(aggregated_points)
self.assertIn(
len(timesteps) - 1,
(
len_aggregated_points,
len_aggregated_points + 1
)
)
unit_length *= max_value_unit
