Python ModelDataFormatter.create_input Beispiele

Beispiel #1

Datei: test_model_data_formatter.py Projekt: svonkleeck/eemeter

def test_basic_hourly(hourly_trace, monkeypatch_temperature_data,
                      mock_isd_weather_source):
    mdf = ModelDataFormatter("H")

    df = mdf.create_input(hourly_trace, mock_isd_weather_source)

    assert all(df.columns == ["energy", "tempF"])
    assert df.index[0] == datetime(2000, 1, 1, 0, tzinfo=pytz.UTC)
    assert df.index[2] == datetime(2000, 1, 1, 2, tzinfo=pytz.UTC)
    assert df.index.freq == 'H'
    assert_allclose(df.energy, [1, 1, 0])
    assert_allclose(df.tempF, [35.617314, 35.607664, 35.598025])

    description = mdf.describe_input(df)
    assert description.get('start_date') == \
        datetime(2000, 1, 1, 0, tzinfo=pytz.UTC)
    assert description.get('end_date') == \
        datetime(2000, 1, 1, 2, tzinfo=pytz.UTC)
    assert description.get('n_rows') == 3

    missing = mdf.get_input_data_mask(df)
    assert missing.shape == (3, )
    assert sum(missing) == 0

    daily = mdf.daily_trace_data(hourly_trace)
    assert daily.shape[0] == 1
    assert_allclose(hourly_trace.data.value.sum(), daily.sum())

Beispiel #2

Datei: test_model_data_formatter.py Projekt: svonkleeck/eemeter

def test_basic_daily(daily_trace, monkeypatch_temperature_data,
                     mock_isd_weather_source):
    mdf = ModelDataFormatter("D")

    df = mdf.create_input(daily_trace, mock_isd_weather_source)

    assert all(df.columns == ["energy", "tempF"])
    assert df.index[0] == datetime(2000, 1, 1, tzinfo=pytz.UTC)
    assert df.index[2] == datetime(2000, 1, 3, tzinfo=pytz.UTC)
    assert df.index.freq == 'D'
    assert_allclose(df.energy, [1, 1, 0.])
    assert_allclose(df.tempF, [35.507344, 35.282373, 35.064391])

    description = mdf.describe_input(df)
    assert description.get('start_date') == \
        datetime(2000, 1, 1, tzinfo=pytz.UTC)
    assert description.get('end_date') == \
        datetime(2000, 1, 3, tzinfo=pytz.UTC)
    assert description.get('n_rows') == 3

    missing = mdf.get_input_data_mask(df)
    assert missing.shape[0] == 3
    assert missing.sum() == 0

    daily = mdf.daily_trace_data(daily_trace)
    assert daily.shape[0] == 3
    assert_allclose(daily_trace.data.value.sum(), daily.sum())

Beispiel #3

Datei: test_model_data_formatter.py Projekt: tsennott/eemeter

def test_basic_15min_to_daily(fifteen_min_trace, mock_isd_weather_source):
    mdf = ModelDataFormatter("D")

    df = mdf.create_input(fifteen_min_trace, mock_isd_weather_source)

    assert all(df.columns == ["energy", "tempF"])
    assert df.index[0] == datetime(2011, 1, 1, 0, tzinfo=pytz.UTC)
    assert df.index.freq == 'D'
    assert df.energy.shape == (100, )
    assert df.tempF.shape == (100, )

Beispiel #4

Datei: test_model_data_formatter.py Projekt: tsennott/eemeter

def test_basic_hourly_to_daily(hourly_trace, mock_isd_weather_source):
    mdf = ModelDataFormatter("D")

    df = mdf.create_input(hourly_trace, mock_isd_weather_source)

    assert all(df.columns == ["energy", "tempF"])
    assert df.index[0] == datetime(2000, 1, 1, 0, tzinfo=pytz.UTC)
    assert df.index.freq == 'D'
    assert_allclose(df.energy, [2])
    assert_allclose(df.tempF, [35.507344])

Beispiel #5

Datei: test_model_data_formatter.py Projekt: brandonwillard/eemeter

def test_basic_hourly_to_daily(hourly_trace, mock_isd_weather_source):
    mdf = ModelDataFormatter("D")

    df = mdf.create_input(hourly_trace, mock_isd_weather_source)

    assert all(df.columns == ["energy", "tempF"])
    assert df.index[0] == datetime(2000, 1, 1, 0, tzinfo=pytz.UTC)
    assert df.index.freq == 'D'
    assert_allclose(df.energy, [2])
    assert_allclose(df.tempF, [32.])

Beispiel #6

def test_basic_hourly(hourly_trace, mock_isd_weather_source):
    mdf = ModelDataFormatter("H")

    df = mdf.create_input(hourly_trace, mock_isd_weather_source)

    assert all(df.columns == ["energy", "tempF"])
    assert df.index[0] == datetime(2000, 1, 1, 0, tzinfo=pytz.UTC)
    assert df.index[2] == datetime(2000, 1, 1, 2, tzinfo=pytz.UTC)
    assert df.index.freq == 'H'
    assert_allclose(df.energy, [1, 1, np.nan])
    assert_allclose(df.tempF, [32., 32., 32.])

    description = mdf.describe_input(df)
    assert description.get('start_date') == \
        datetime(2000, 1, 1, 0, tzinfo=pytz.UTC)
    assert description.get('end_date') == \
        datetime(2000, 1, 1, 2, tzinfo=pytz.UTC)
    assert description.get('n_rows') == 3

Beispiel #7

def input_df(monkeypatch_temperature_data, daily_trace,
             mock_isd_weather_source):
    mdf = ModelDataFormatter("D")
    return mdf.create_input(daily_trace, mock_isd_weather_source)

Beispiel #8

Datei: test_model_data_formatter.py Projekt: svonkleeck/eemeter

def test_daily_to_hourly_fails(daily_trace, monkeypatch_temperature_data,
                               mock_isd_weather_source):
    mdf = ModelDataFormatter("H")

    with pytest.raises(ValueError):
        mdf.create_input(daily_trace, mock_isd_weather_source)

Beispiel #9

Datei: test_hourly_load_profile_model.py Projekt: zimmemit/eemeter

def input_df(mock_isd_weather_source, hourly_trace):
    mdf = ModelDataFormatter("H")
    return mdf.create_input(hourly_trace, mock_isd_weather_source)

Beispiel #10

Datei: test_model_data_formatter.py Projekt: brandonwillard/eemeter

def test_daily_to_hourly_fails(daily_trace, mock_isd_weather_source):
    mdf = ModelDataFormatter("H")

    with pytest.raises(ValueError):
        mdf.create_input(daily_trace, mock_isd_weather_source)

Beispiel #11

Datei: test_seasonal_model.py Projekt: brandonwillard/eemeter

def input_df(mock_isd_weather_source, daily_trace):
    mdf = ModelDataFormatter("D")
    return mdf.create_input(daily_trace, mock_isd_weather_source)