def test_interpolate(self):
data = [(1379487600, 2.0), (1379488500, 4.0), (1379489400, 6.0), (1379490300, 8.0)]
expected = [(1379488050, 3.0), (1379488950, 5.0), (1379489850, 7.0)]
series = Series(data)
out = series.data(start_at=1379488050, end_at=1379489850, step_size=900)
assert out == expected
def test_interpolate(self):
data = [(1379487600, 2.0), (1379488500, 4.0), (1379489400, 6.0),
(1379490300, 8.0)]
expected = [(1379488050, 3.0), (1379488950, 5.0), (1379489850, 7.0)]
series = Series(data)
out = series.data(start_at=1379488050,
end_at=1379489850,
step_size=900)
assert out == expected
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_named_exculsions(self):
series = Series(self.get_temp_data_filepath(), temp_units='C')
series.add_named_exclusion("US_HOLIDAYS")
assert len(series.data()) < len(series.series)
def test_temp_flag_is_not_f(self):
series = Series(self.get_temp_data_filepath(), temp_units='C')
assert series.is_farenheit() == False
def test_temp_flag_default_is_f(self):
series = Series(self.get_kw_data_filepath())
assert series.is_farenheit()
def test_data_slice(self):
series = Series(self.dummy_data())
assert len(series.data(1379488500, 1379490300)) == 3
def test_exclusion(self):
series = Series(self.dummy_data())
series.add_exclusion(1379488500, 1379490300)
assert len(series.data()) == 2
def test_validity(self):
series = Series(self.get_kw_data_filepath())
assert series.valid()
def test_named_exculsions(self):
series = Series(self.get_temp_data_filepath(), temp_units='C')
series.add_named_exclusion("US_HOLIDAYS")
assert len(series.data()) < len(series.series)
def test_temp_flag_is_not_f(self):
series = Series(self.get_temp_data_filepath(), temp_units='C')
assert series.is_farenheit() == False
def test_temp_flag_default_is_f(self):
series = Series(self.get_kw_data_filepath())
assert series.is_farenheit()
def test_data_slice(self):
series = Series(self.dummy_data())
assert len(series.data(1379488500, 1379490300)) == 3
def test_exclusion(self):
series = Series(self.dummy_data())
series.add_exclusion(1379488500, 1379490300)
assert len(series.data()) == 2
def test_validity(self):
series = Series(self.get_kw_data_filepath())
assert series.valid()
