Source code for gridstatus.tests.test_ercot

import pandas as pd
import pytest

import gridstatus


@pytest.mark.skip(reason="takes too long to run")
[docs]def test_ercot_get_historical_rtm_spp():
    rtm = gridstatus.Ercot().get_historical_rtm_spp(2020)
    assert isinstance(rtm, pd.DataFrame)
    assert len(rtm) > 0


[docs]def test_ercot_get_fuel_mix():

    # today
    iso = gridstatus.Ercot()
    df = iso.get_fuel_mix("today")
    assert df.shape[0] >= 0
    assert df.columns.tolist() == ["Time", "Solar", "Wind", "Other"]

    # latest
    df = iso.get_fuel_mix("latest").mix
    assert df.shape[0] >= 0
    assert df.columns.tolist() == ["Time", "Solar", "Wind", "Other"]


[docs]def test_ercot_get_as_prices():
    as_cols = [
        "Time",
        "Market",
        "Non-Spinning Reserves",
        "Regulation Down",
        "Regulation Up",
        "Responsive Reserves",
    ]

    # today
    iso = gridstatus.Ercot()
    today = pd.Timestamp.now().date()
    df = iso.get_as_prices(today)
    assert df.shape[0] >= 0
    assert df.columns.tolist() == as_cols
    assert df["Time"].unique()[0].date() == today

    date = today - pd.Timedelta(days=3)
    df = iso.get_as_prices(date)
    assert df.shape[0] >= 0
    assert df.columns.tolist() == as_cols
    assert df["Time"].unique()[0].date() == date