Skip to content

Commit

Permalink
feat: enable retrieval of pf and dcline_pf from MockAnalyze (#502)
Browse files Browse the repository at this point in the history
  • Loading branch information
danielolsen authored Jun 15, 2021
1 parent defd713 commit 37e6394
Showing 1 changed file with 14 additions and 0 deletions.
14 changes: 14 additions & 0 deletions powersimdata/tests/mock_analyze.py
Original file line number Diff line number Diff line change
Expand Up @@ -33,6 +33,7 @@ def __init__(
lmp=None,
pf=None,
pg=None,
dcline_pf=None,
storage_pg=None,
solar=None,
wind=None,
Expand Down Expand Up @@ -60,6 +61,7 @@ def __init__(
self.demand = _ensure_ts_index(demand)
self.lmp = _ensure_ts_index(lmp)
self.pf = _ensure_ts_index(pf)
self.dcline_pf = _ensure_ts_index(dcline_pf)
self.pg = _ensure_ts_index(pg)
self.storage_pg = _ensure_ts_index(storage_pg)
self.solar = _ensure_ts_index(solar)
Expand Down Expand Up @@ -103,6 +105,18 @@ def get_lmp(self):
"""
return self.lmp

def get_pf(self):
"""Get PF.
:return: (pandas.DataFrame) -- dummy pf
"""
return self.pf

def get_dcline_pf(self):
"""Get PF_DCLINE.
:return: (pandas.DataFrame) -- dummy pf_dcline
"""
return self.dcline_pf

def get_pg(self):
"""Get PG.
:return: (pandas.DataFrame) -- dummy pg
Expand Down

0 comments on commit 37e6394

Please sign in to comment.