import pytest
from gridstatus import NYISO, Markets
from gridstatus.tests.base_test_iso import BaseTestISO
from gridstatus.tests.decorators import with_markets
[docs]class TestNYISO(BaseTestISO):
iso = NYISO()
""""get_capacity_prices"""
[docs] def test_get_capacity_prices(self):
# test 2022, 2023, and today
df = self.iso.get_capacity_prices(date="Dec 1, 2022", verbose=True)
assert not df.empty, "DataFrame came back empty"
df = self.iso.get_capacity_prices(date="Jan 1, 2023", verbose=True)
assert not df.empty, "DataFrame came back empty"
df = self.iso.get_capacity_prices(date="today", verbose=True)
assert not df.empty, "DataFrame came back empty"
"""get_fuel_mix"""
[docs] def test_get_fuel_mix_date_range(self):
df = self.iso.get_fuel_mix(start="Aug 1, 2022", end="Oct 22, 2022")
assert df.shape[0] >= 0
"""get_generators"""
[docs] def test_get_generators(self):
df = self.iso.get_generators()
columns = [
"Generator Name",
"PTID",
"Subzone",
"Zone",
"Latitude",
"Longitude",
]
assert set(df.columns).issuperset(set(columns))
assert df.shape[0] >= 0
"""get_lmp"""
@with_markets(
Markets.DAY_AHEAD_HOURLY,
Markets.REAL_TIME_5_MIN,
)
[docs] def test_get_lmp_historical(self, market):
super().test_get_lmp_historical(market=market)
[docs] def test_get_lmp_historical_with_range(self):
start = "2021-12-01"
end = "2022-2-02"
df = self.iso.get_lmp(
start=start,
end=end,
market=Markets.REAL_TIME_5_MIN,
)
assert df.shape[0] >= 0
@with_markets(
Markets.DAY_AHEAD_HOURLY,
Markets.REAL_TIME_5_MIN,
)
[docs] def test_get_lmp_latest(self, market):
super().test_get_lmp_latest(market=market)
[docs] def test_get_lmp_location_type_parameter(self):
date = "2022-06-09"
df_zone = self.iso.get_lmp(
date=date,
market=Markets.DAY_AHEAD_HOURLY,
location_type="zone",
)
assert (df_zone["Location Type"] == "Zone").all()
df_gen = self.iso.get_lmp(
date=date,
market=Markets.DAY_AHEAD_HOURLY,
location_type="generator",
)
assert (df_gen["Location Type"] == "Generator").all()
df_zone = self.iso.get_lmp(
date="today",
market=Markets.DAY_AHEAD_HOURLY,
location_type="zone",
)
assert (df_zone["Location Type"] == "Zone").all()
df_gen = self.iso.get_lmp(
date="today",
market=Markets.DAY_AHEAD_HOURLY,
location_type="generator",
)
assert (df_gen["Location Type"] == "Generator").all()
df_zone = self.iso.get_lmp(
date="latest",
market=Markets.DAY_AHEAD_HOURLY,
location_type="zone",
)
assert (df_zone["Location Type"] == "Zone").all()
df_gen = self.iso.get_lmp(
date="latest",
market=Markets.DAY_AHEAD_HOURLY,
location_type="generator",
)
assert (df_gen["Location Type"] == "Generator").all()
with pytest.raises(ValueError):
self.iso.get_lmp(
date="latest",
market=Markets.DAY_AHEAD_HOURLY,
location_type="dummy",
)
@with_markets(
Markets.DAY_AHEAD_HOURLY,
Markets.REAL_TIME_5_MIN,
)
[docs] def test_get_lmp_today(self, market):
super().test_get_lmp_today(market=market)
"""get_loads"""
[docs] def test_get_loads(self):
df = self.iso.get_loads()
columns = [
"Load Name",
"PTID",
"Subzone",
"Zone",
]
assert set(df.columns) == set(columns)
assert df.shape[0] >= 0
"""get_status"""
[docs] def test_get_status_historical_status(self):
date = "20220609"
status = self.iso.get_status(date)
self._check_status(status)
start = "2022-05-01"
end = "2022-10-02"
status = self.iso.get_status(start=start, end=end)
self._check_status(status)
"""get_storage"""
[docs] def test_get_storage_historical(self):
with pytest.raises(NotImplementedError):
super().test_get_storage_historical()
[docs] def test_get_storage_today(self):
with pytest.raises(NotImplementedError):
super().test_get_storage_today()
[docs] def test_various_edt_to_est(self):
# number of rows hardcoded based on when this test was written. should stay same
date = "Nov 7, 2021"
df = self.iso.get_status(date=date)
assert df.shape[0] >= 1
df = self.iso.get_fuel_mix(date=date)
assert df.shape[0] >= 307
df = self.iso.get_load_forecast(date=date)
assert df.shape[0] >= 145
df = self.iso.get_lmp(date=date, market=Markets.REAL_TIME_5_MIN)
assert df.shape[0] >= 4605
df = self.iso.get_lmp(date=date, market=Markets.DAY_AHEAD_HOURLY)
assert df.shape[0] >= 375
df = self.iso.get_load(date=date)
assert df.shape[0] >= 307
[docs] def test_various_est_to_edt(self):
# number of rows hardcoded based on when this test was written. should stay same
date = "March 14, 2021"
df = self.iso.get_status(date=date)
assert df.shape[0] >= 5
df = self.iso.get_lmp(date=date, market=Markets.REAL_TIME_5_MIN)
assert df.shape[0] >= 4215
df = self.iso.get_lmp(date=date, market=Markets.DAY_AHEAD_HOURLY)
assert df.shape[0] >= 345
df = self.iso.get_load_forecast(date=date)
assert df.shape[0] >= 143
df = self.iso.get_fuel_mix(date=date)
assert df.shape[0] >= 281
df = self.iso.get_load(date=date)
assert df.shape[0] >= 281
@staticmethod
def _check_status(df):
assert set(df.columns) == set(
["Time", "Status", "Notes"],
)
