def test_diff(self):
from loadshape.utils import get_timezone
l_data = [(1379487600, 5.0), (1379488500, 5.0), (1379489400, 5.0),
(1379490300, 5.0), (1379491200, 5.0)]
b_data = [(1379487600, 4.0), (1379488500, 4.0), (1379489400, 4.0),
(1379490300, 4.0), (1379491200, 4.0)]
expected_kw_diff = [(1379488500, 1), (1379489400, 1), (1379490300, 1),
(1379491200, 1)]
expected_cumulative_kwh_diff = [(1379487600, 0), (1379488500, 0.25),
(1379489400, 0.5), (1379490300, 0.75),
(1379491200, 1.0)]
expected_cumulative_kwh_base = [(1379487600, 0), (1379488500, 1.0),
(1379489400, 2.0), (1379490300, 3.0),
(1379491200, 4.0)]
b = Loadshape(l_data, timezone='America/Los_Angeles', log_level=30)
b.baseline_series = Series(b_data, get_timezone('America/Los_Angeles'))
kw_diff, kw_base, cumulative_kwh_diff, cumulative_kwh_base = b.diff()
assert kw_diff.data() == expected_kw_diff
assert cumulative_kwh_diff.data() == expected_cumulative_kwh_diff
assert kw_base.data() == b_data[1:]
assert cumulative_kwh_base.data() == expected_cumulative_kwh_base
def test_diff(self):
l_data = [(1379487600, 5.0), (1379488500, 5.0), (1379489400, 5.0), (1379490300, 5.0), (1379491200, 5.0)]
b_data = [(1379487600, 4.0), (1379488500, 4.0), (1379489400, 4.0), (1379490300, 4.0), (1379491200, 4.0)]
expected_kw_diff = [(1379488500, 1), (1379489400, 1), (1379490300, 1), (1379491200, 1)]
expected_cumulative_kwh_diff = [(1379487600, 0), (1379488500, 0.25), (1379489400, 0.5), (1379490300, 0.75), (1379491200, 1.0)]
expected_cumulative_kwh_base = [(1379487600, 0), (1379488500, 1.0), (1379489400, 2.0), (1379490300, 3.0), (1379491200, 4.0)]
b = Loadshape(l_data, timezone='America/Los_Angeles', log_level=30)
b.baseline_series = Series(b_data)
kw_diff, kw_base, cumulative_kwh_diff, cumulative_kwh_base = b.diff()
assert kw_diff.data() == expected_kw_diff
assert cumulative_kwh_diff.data() == expected_cumulative_kwh_diff
assert kw_base.data() == b_data[1:]
assert cumulative_kwh_base.data() == expected_cumulative_kwh_base
