Merge pull request #422 from Breakthrough-Energy/daniel/create_get_bu…

…s_demand

fix: ensure that get_bus_demand works in Create state

powersimdata/input/input_data.py

@@ -144,8 +144,8 @@ def get_bus_demand(scenario_info, grid):
     :param powersimdata.input.grid.Grid grid: grid to construct bus demand for.
     :return: (*pandas.DataFrame*) -- data frame of demand.
     """
-    demand = InputData().get_data(scenario_info, "demand")
     bus = grid.bus
+    demand = InputData().get_data(scenario_info, "demand")[bus.zone_id.unique()]
     bus["zone_Pd"] = bus.groupby("zone_id")["Pd"].transform("sum")
     bus["zone_share"] = bus["Pd"] / bus["zone_Pd"]
     zone_bus_shares = pd.DataFrame(

powersimdata/scenario/create.py

@@ -149,6 +149,7 @@ def get_bus_demand(self):
 
         :return: (*pandas.DataFrame*) -- data frame of demand (hour, bus).
         """
+        self._update_scenario_info()
         grid = self.get_grid()
         return get_bus_demand(self._scenario_info, grid)
 

powersimdata/scenario/tests/test_create.py

@@ -0,0 +1,11 @@
+import pytest
+
+from powersimdata.scenario.scenario import Scenario
+
+
+@pytest.mark.ssh
+def test_get_bus_demand():
+    scenario = Scenario("")
+    scenario.state.set_builder(interconnect="Texas")
+    scenario.state.builder.set_base_profile("demand", "vJan2021")
+    scenario.state.get_bus_demand()