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