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