From 37e6394b4d11e34af88f541fc570842912eeb63c Mon Sep 17 00:00:00 2001
From: danielolsen <danielolsen@users.noreply.github.com>
Date: Tue, 15 Jun 2021 15:20:21 -0700
Subject: [PATCH] feat: enable retrieval of pf and dcline_pf from MockAnalyze
 (#502)

---
 powersimdata/tests/mock_analyze.py | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/powersimdata/tests/mock_analyze.py b/powersimdata/tests/mock_analyze.py
index d87ad9e68..c00bd6d2d 100644
--- a/powersimdata/tests/mock_analyze.py
+++ b/powersimdata/tests/mock_analyze.py
@@ -33,6 +33,7 @@ def __init__(
         lmp=None,
         pf=None,
         pg=None,
+        dcline_pf=None,
         storage_pg=None,
         solar=None,
         wind=None,
@@ -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)
@@ -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