def test_create_list_no_labels(trace1, trace2): traces = [trace1, trace2] ets = EnergyTraceSet(traces) assert len(list(ets.itertraces())) == 2 assert list(ets.itertraces())[0][0] in ['0', '1'] assert list(ets.itertraces())[1][0] in ['0', '1']
def test_create_list(trace1, trace2): traces = [trace1, trace2] labels = ["t1", "t2"] ets = EnergyTraceSet(traces, labels) assert len(list(ets.itertraces())) == 2 assert list(ets.itertraces())[0][0] in ['t1', 't2'] assert list(ets.itertraces())[1][0] in ['t1', 't2']
def eemeter_project(self): cm_set = self.consumptionmetadata_set.all() consumption = [cm.eemeter_consumption_data() for cm in cm_set] consumption_metadata_ids = [cm.id for cm in cm_set] energy_trace_set = EnergyTraceSet(consumption, consumption_metadata_ids) interventions = [self._eemeter_intervention()] site = self._eemeter_location() return EEMeterProject(energy_trace_set, interventions, site)
def trace_set(): columns = { "value": [1, 1, 1, 1, np.nan], "estimated": [False, False, False, False, False] } column_names = ["value", "estimated"] index = pd.date_range('2000-01-01', periods=5, freq='D') data = pd.DataFrame(columns, index=index, columns=column_names) trace = EnergyTrace("ELECTRICITY_ON_SITE_GENERATION_UNCONSUMED", data=data, unit="KWH") return EnergyTraceSet([trace], ["trace"])
def placeholder_trace_set(): trace = EnergyTrace("ELECTRICITY_ON_SITE_GENERATION_UNCONSUMED", placeholder=True) return EnergyTraceSet([trace], ["trace"])
def energy_trace_set(daily_data): energy_trace_set = EnergyTraceSet([ EnergyTrace('ELECTRICITY_CONSUMPTION_SUPPLIED', data=daily_data, unit='kWh'), ]) return energy_trace_set
def project_bad_zip(): ets = EnergyTraceSet({}) interventions = [] site = ZIPCodeSite("00000") project = Project(ets, interventions, site) return project
def project(): ets = EnergyTraceSet({}) interventions = [] site = ZIPCodeSite("91104") project = Project(ets, interventions, site) return project
def test_repr(trace1, trace2): traces = [trace1, trace2] ets = EnergyTraceSet(traces) assert str(ets).startswith("EnergyTraceSet(traces={") assert str(ets).endswith("})")
def test_create_list_double_labels(trace1, trace2): traces = [trace1, trace2] labels = ["t1", "t1"] with pytest.raises(ValueError): EnergyTraceSet(traces, labels)
def test_create_dict_with_labels(trace1, trace2): trace_dict = {"t1": trace1, "t2": trace2} with pytest.warns(UserWarning): EnergyTraceSet(trace_dict, labels=["t1", "t2"])
def test_create_dict(trace1, trace2): trace_dict = {"t1": trace1, "t2": trace2} ets = EnergyTraceSet(trace_dict) assert len(list(ets.itertraces())) == 2 assert list(ets.itertraces())[0][0] in ['t1', 't2'] assert list(ets.itertraces())[1][0] in ['t1', 't2']
def energy_trace_set(): return EnergyTraceSet([])