def test_create_caltrack_billing_design_matrix_partial_empty_temp(
il_electricity_cdd_hdd_billing_monthly):
meter_data = il_electricity_cdd_hdd_billing_monthly["meter_data"]
temperature_data = il_electricity_cdd_hdd_billing_monthly[
"temperature_data"][:200]
design_matrix = create_caltrack_billing_design_matrix(
meter_data[:10], temperature_data)
assert "n_days_kept" in design_matrix.columns
def baseline_model_billing_single_record(il_electricity_cdd_hdd_billing_monthly):
baseline_meter_data = il_electricity_cdd_hdd_billing_monthly["meter_data"][-2:]
temperature_data = il_electricity_cdd_hdd_billing_monthly["temperature_data"]
blackout_start_date = il_electricity_cdd_hdd_billing_monthly["blackout_start_date"]
baseline_data = create_caltrack_billing_design_matrix(
baseline_meter_data, temperature_data
)
model_results = fit_caltrack_usage_per_day_model(
baseline_data, use_billing_presets=True, weights_col="n_days_kept"
)
return model_results
def reporting_model_billing(il_electricity_cdd_hdd_billing_monthly):
meter_data = il_electricity_cdd_hdd_billing_monthly["meter_data"]
meter_data.value = meter_data.value - 50
temperature_data = il_electricity_cdd_hdd_billing_monthly[
"temperature_data"]
blackout_start_date = il_electricity_cdd_hdd_billing_monthly[
"blackout_start_date"]
baseline_meter_data, warnings = get_baseline_data(meter_data,
end=blackout_start_date)
baseline_data = create_caltrack_billing_design_matrix(
baseline_meter_data, temperature_data)
model_results = fit_caltrack_usage_per_day_model(baseline_data,
use_billing_presets=True,
weights_col="n_days_kept")
return model_results
def test_create_caltrack_billing_design_matrix(
il_electricity_cdd_hdd_billing_monthly):
meter_data = il_electricity_cdd_hdd_billing_monthly["meter_data"]
temperature_data = il_electricity_cdd_hdd_billing_monthly[
"temperature_data"]
design_matrix = create_caltrack_billing_design_matrix(
meter_data[:10], temperature_data)
assert design_matrix.shape == (10, 128)
assert sorted(design_matrix.columns) == [
"cdd_30",
"cdd_31",
"cdd_32",
"cdd_33",
"cdd_34",
"cdd_35",
"cdd_36",
"cdd_37",
"cdd_38",
"cdd_39",
"cdd_40",
"cdd_41",
"cdd_42",
"cdd_43",
"cdd_44",
"cdd_45",
"cdd_46",
"cdd_47",
"cdd_48",
"cdd_49",
"cdd_50",
"cdd_51",
"cdd_52",
"cdd_53",
"cdd_54",
"cdd_55",
"cdd_56",
"cdd_57",
"cdd_58",
"cdd_59",
"cdd_60",
"cdd_61",
"cdd_62",
"cdd_63",
"cdd_64",
"cdd_65",
"cdd_66",
"cdd_67",
"cdd_68",
"cdd_69",
"cdd_70",
"cdd_71",
"cdd_72",
"cdd_73",
"cdd_74",
"cdd_75",
"cdd_76",
"cdd_77",
"cdd_78",
"cdd_79",
"cdd_80",
"cdd_81",
"cdd_82",
"cdd_83",
"cdd_84",
"cdd_85",
"cdd_86",
"cdd_87",
"cdd_88",
"cdd_89",
"cdd_90",
"hdd_30",
"hdd_31",
"hdd_32",
"hdd_33",
"hdd_34",
"hdd_35",
"hdd_36",
"hdd_37",
"hdd_38",
"hdd_39",
"hdd_40",
"hdd_41",
"hdd_42",
"hdd_43",
"hdd_44",
"hdd_45",
"hdd_46",
"hdd_47",
"hdd_48",
"hdd_49",
"hdd_50",
"hdd_51",
"hdd_52",
"hdd_53",
"hdd_54",
"hdd_55",
"hdd_56",
"hdd_57",
"hdd_58",
"hdd_59",
"hdd_60",
"hdd_61",
"hdd_62",
"hdd_63",
"hdd_64",
"hdd_65",
"hdd_66",
"hdd_67",
"hdd_68",
"hdd_69",
"hdd_70",
"hdd_71",
"hdd_72",
"hdd_73",
"hdd_74",
"hdd_75",
"hdd_76",
"hdd_77",
"hdd_78",
"hdd_79",
"hdd_80",
"hdd_81",
"hdd_82",
"hdd_83",
"hdd_84",
"hdd_85",
"hdd_86",
"hdd_87",
"hdd_88",
"hdd_89",
"hdd_90",
"meter_value",
"n_days_dropped",
"n_days_kept",
"temperature_mean",
"temperature_not_null",
"temperature_null",
]
assert round(design_matrix.sum().sum(), 2) == 19365.12