def test_average_larger(self):
time_interval = timedelta(seconds=60)
timeseries = [
[
datetime.strptime('2017-11-16 13:50:00', '%Y-%m-%d %H:%M:%S'),
4.0
],
[
datetime.strptime('2017-11-16 13:55:00', '%Y-%m-%d %H:%M:%S'),
5.0
],
[
datetime.strptime('2017-11-16 14:01:00', '%Y-%m-%d %H:%M:%S'),
6.0
],
[
datetime.strptime('2017-11-16 14:06:00', '%Y-%m-%d %H:%M:%S'),
-999
],
[
datetime.strptime('2017-11-16 14:11:00', '%Y-%m-%d %H:%M:%S'),
7.0
],
]
new_timeseries = \
InterpolationStrategy.get_strategy_for_larger(InterpolationStrategy.Average)(timeseries, time_interval)
for t in new_timeseries:
print(t)
self.assertEqual(len(new_timeseries), 22)
self.assertListEqual(new_timeseries[19], [
datetime.strptime('2017-11-16 14:09:00', '%Y-%m-%d %H:%M:%S'), -999
])
def test_spread_fill(self):
step1 = [
datetime.strptime('2017-11-15 08:20:00', '%Y-%m-%d %H:%M:%S'), 2.0
]
step2 = [
datetime.strptime('2017-11-15 08:30:00', '%Y-%m-%d %H:%M:%S'), 2.0
]
new_timeseries = InterpolationStrategy.spread_fill(
step1, step2, timedelta(minutes=5))
print(new_timeseries)
self.assertEqual(len(new_timeseries), 2)
self.assertListEqual(new_timeseries[1], [
datetime.strptime('2017-11-15 08:25:00', '%Y-%m-%d %H:%M:%S'), 1.0
])
step1 = [
datetime.strptime('2017-11-15 08:20:00', '%Y-%m-%d %H:%M:%S'), 2.0
]
step2 = [
datetime.strptime('2017-11-15 08:40:00', '%Y-%m-%d %H:%M:%S'), 2.0
]
new_timeseries = InterpolationStrategy.spread_fill(
step1, step2, timedelta(minutes=5))
print(new_timeseries)
self.assertEqual(len(new_timeseries), 4)
self.assertListEqual(new_timeseries[1], [
datetime.strptime('2017-11-15 08:25:00', '%Y-%m-%d %H:%M:%S'), 0.5
])
step1 = [
datetime.strptime('2017-11-15 08:20:00', '%Y-%m-%d %H:%M:%S'), 2.0
]
step2 = [
datetime.strptime('2017-11-15 08:40:00', '%Y-%m-%d %H:%M:%S'), 2.0
]
new_timeseries = InterpolationStrategy.spread_fill(
step1, step2, timedelta(minutes=5), -999)
print(new_timeseries)
self.assertEqual(len(new_timeseries), 4)
self.assertListEqual(new_timeseries[1], [
datetime.strptime('2017-11-15 08:25:00', '%Y-%m-%d %H:%M:%S'), -999
])
def test_average_smaller(self):
time_interval = timedelta(seconds=60)
# NOTE: Assume that there isn't any gaps in the timeseries
# Gaps need to be fulfill before going to feed into the function
timeseries = [
[
datetime.strptime('2017-11-15 08:20:29', '%Y-%m-%d %H:%M:%S'),
1.0
],
[
datetime.strptime('2017-11-15 08:20:45', '%Y-%m-%d %H:%M:%S'),
2.0
],
[
datetime.strptime('2017-11-15 08:21:01', '%Y-%m-%d %H:%M:%S'),
3.0
],
[
datetime.strptime('2017-11-15 08:21:17', '%Y-%m-%d %H:%M:%S'),
4.0
],
[
datetime.strptime('2017-11-15 08:21:33', '%Y-%m-%d %H:%M:%S'),
5.0
],
[
datetime.strptime('2017-11-15 08:21:49', '%Y-%m-%d %H:%M:%S'),
6.0
],
[
datetime.strptime('2017-11-15 08:22:05', '%Y-%m-%d %H:%M:%S'),
7.0
],
[
datetime.strptime('2017-11-15 08:22:21', '%Y-%m-%d %H:%M:%S'),
8.0
],
[
datetime.strptime('2017-11-15 08:22:37', '%Y-%m-%d %H:%M:%S'),
9.0
],
[
datetime.strptime('2017-11-15 08:22:53', '%Y-%m-%d %H:%M:%S'),
10.0
],
]
new_timeseries = \
InterpolationStrategy.get_strategy_for_smaller(InterpolationStrategy.Average)(timeseries, time_interval)
print('Result::')
for t in new_timeseries:
print(t)
print('\n')
self.assertEqual(len(new_timeseries), 3)
self.assertListEqual(new_timeseries[1], [
datetime.strptime('2017-11-15 08:21:29', '%Y-%m-%d %H:%M:%S'), 6.5
])
timeseries2 = [
[
datetime.strptime('2017-11-15 00:00:00', '%Y-%m-%d %H:%M:%S'),
1.0
],
[
datetime.strptime('2017-11-15 00:00:16', '%Y-%m-%d %H:%M:%S'),
2.0
],
[
datetime.strptime('2017-11-15 00:00:32', '%Y-%m-%d %H:%M:%S'),
-999
],
[
datetime.strptime('2017-11-15 00:00:48', '%Y-%m-%d %H:%M:%S'),
4.0
],
[
datetime.strptime('2017-11-15 00:01:04', '%Y-%m-%d %H:%M:%S'),
-999
],
[
datetime.strptime('2017-11-15 00:01:20', '%Y-%m-%d %H:%M:%S'),
6.0
],
[
datetime.strptime('2017-11-15 00:01:36', '%Y-%m-%d %H:%M:%S'),
-999
],
[
datetime.strptime('2017-11-15 00:01:52', '%Y-%m-%d %H:%M:%S'),
7.0
],
[
datetime.strptime('2017-11-15 00:02:08', '%Y-%m-%d %H:%M:%S'),
8.0
],
[
datetime.strptime('2017-11-15 00:02:24', '%Y-%m-%d %H:%M:%S'),
9.0
],
]
new_timeseries = \
InterpolationStrategy.get_strategy_for_smaller(InterpolationStrategy.Average)(timeseries2, time_interval)
print('Result::')
for t in new_timeseries:
print(t)
self.assertEqual(len(new_timeseries), 3)
self.assertListEqual(new_timeseries[1], [
datetime.strptime('2017-11-15 00:01:00', '%Y-%m-%d %H:%M:%S'), 6.5
])
def test_summation_smaller(self):
time_interval = timedelta(seconds=60)
timeseries = [
[
datetime.strptime('2017-11-15 08:20:29', '%Y-%m-%d %H:%M:%S'),
1.0
],
[
datetime.strptime('2017-11-15 08:20:45', '%Y-%m-%d %H:%M:%S'),
2.0
],
[
datetime.strptime('2017-11-15 08:21:01', '%Y-%m-%d %H:%M:%S'),
3.0
],
[
datetime.strptime('2017-11-15 08:21:17', '%Y-%m-%d %H:%M:%S'),
4.0
],
[
datetime.strptime('2017-11-15 08:21:33', '%Y-%m-%d %H:%M:%S'),
5.0
],
[
datetime.strptime('2017-11-15 08:21:49', '%Y-%m-%d %H:%M:%S'),
6.0
],
[
datetime.strptime('2017-11-15 08:22:05', '%Y-%m-%d %H:%M:%S'),
7.0
],
[
datetime.strptime('2017-11-15 08:22:21', '%Y-%m-%d %H:%M:%S'),
8.0
],
[
datetime.strptime('2017-11-15 08:22:37', '%Y-%m-%d %H:%M:%S'),
9.0
],
[
datetime.strptime('2017-11-15 08:22:53', '%Y-%m-%d %H:%M:%S'),
10.0
],
]
new_timeseries = \
InterpolationStrategy.get_strategy_for_smaller(InterpolationStrategy.Summation)(timeseries, time_interval)
print('Result::')
for t in new_timeseries:
print(t)
print('\n')
self.assertEqual(len(new_timeseries), 3)
timeseries2 = [
[
datetime.strptime('2017-11-15 00:00:00', '%Y-%m-%d %H:%M:%S'),
1.0
],
[
datetime.strptime('2017-11-15 00:00:16', '%Y-%m-%d %H:%M:%S'),
2.0
],
[
datetime.strptime('2017-11-15 00:00:32', '%Y-%m-%d %H:%M:%S'),
-999
],
[
datetime.strptime('2017-11-15 00:00:48', '%Y-%m-%d %H:%M:%S'),
4.0
],
[
datetime.strptime('2017-11-15 00:01:04', '%Y-%m-%d %H:%M:%S'),
-999
],
[
datetime.strptime('2017-11-15 00:01:20', '%Y-%m-%d %H:%M:%S'),
6.0
],
[
datetime.strptime('2017-11-15 00:01:36', '%Y-%m-%d %H:%M:%S'),
-999
],
[
datetime.strptime('2017-11-15 00:01:52', '%Y-%m-%d %H:%M:%S'),
7.0
],
[
datetime.strptime('2017-11-15 00:02:08', '%Y-%m-%d %H:%M:%S'),
8.0
],
[
datetime.strptime('2017-11-15 00:02:24', '%Y-%m-%d %H:%M:%S'),
9.0
],
]
new_timeseries = \
InterpolationStrategy.get_strategy_for_smaller(InterpolationStrategy.Summation)(timeseries2, time_interval)
print('Result::')
for t in new_timeseries:
print(t)
self.assertEqual(len(new_timeseries), 3)