Move modules

powersimdata/design/scenario_info.py

@@ -1,6 +1,6 @@
 import warnings
 
-from prereise.gather.constants import abv2state, state2loadzone, \
+from powersimdata.utility.constants import abv2state, state2loadzone, \
     interconnect2loadzone
 
 

powersimdata/design/tests/test_scenario_info.py

@@ -2,7 +2,7 @@
 import pandas as pd
 
 from powersimdata.design.scenario_info import ScenarioInfo
-from postreise.tests.mock_scenario import MockScenario
+from powersimdata.tests.mock_scenario import MockScenario
 
 mock_plant = {
     'plant_id': [101, 102, 103, 104, 105, 106],

powersimdata/design/tests/test_transmission.py

@@ -4,7 +4,7 @@
 import pandas as pd
 
 from powersimdata.tests.mock_grid import MockGrid
-from postreise.tests.mock_scenario import MockScenario
+from powersimdata.tests.mock_scenario import MockScenario
 from powersimdata.tests.mock_change_table import MockChangeTable
 from powersimdata.design.transmission import (
     _find_branches_connected_to_bus, _find_first_degree_branches,

powersimdata/design/transmission.py

@@ -1,7 +1,7 @@
 import numpy as np
 import pandas as pd
 
-from postreise.analyze.transmission.distance import haversine
+from powersimdata.utility.distance import haversine
 
 
 def _find_branches_connected_to_bus(branch, bus_id):

powersimdata/tests/mock_analyze.py

@@ -0,0 +1,96 @@
+from powersimdata.scenario.analyze import Analyze
+from powersimdata.tests.mock_grid import MockGrid
+
+
+class MockAnalyze:
+    def __init__(self, grid_attrs, congl=None, congu=None, ct=None,
+                 demand=None, lmp=None, pg=None, solar=None, wind=None, hydro=None):
+        """Constructor.
+
+        :param dict grid_attrs: fields to be added to grid.
+        :param pandas.DataFrame congl: dummy congl
+        :param pandas.DataFrame congu: dummy congu
+        :param dict ct: dummy ct
+        :param pandas.DataFrame demand: dummy demand
+        :param pandas.DataFrame lmp: dummy lmp
+        :param pandas.DataFrame pg: dummy pg
+        :param pandas.DataFrame solar: dummy solar
+        :param pandas.DataFrame wind: dummy wind
+        :param pandas.DataFrame hydro: dummy hydro
+        """
+        self.grid = MockGrid(grid_attrs)
+        self.congl = congl
+        self.congu = congu
+        self.ct = ct if ct is not None else {}
+        self.demand = demand
+        self.lmp = lmp
+        self.pg = pg
+        self.solar = solar
+        self.wind = wind
+        self.hydro = hydro
+        self.name = 'analyze'
+
+    def get_congl(self):
+        """Get congl.
+        :return: (pandas.DataFrame) -- dummy congl
+        """
+        return self.congl
+
+    def get_congu(self):
+        """Get congu.
+        :return: (pandas.DataFrame) -- dummy congu
+        """
+        return self.congu
+
+    def get_ct(self):
+        """Get ct.
+        :return: (Dict) -- dummy ct
+        """
+        return self.ct
+
+    def get_demand(self, original=None):
+        """Get demand.
+        :return: (pandas.DataFrame) -- dummy demand
+        """
+        return self.demand
+
+    def get_grid(self):
+        """Get grid.
+        :return: (MockGrid) -- mock grid
+        """
+        return self.grid
+
+    def get_lmp(self):
+        """Get lmp.
+        :return: (pandas.DataFrame) -- dummy lmp
+        """
+        return self.lmp
+
+    def get_pg(self):
+        """Get PG.
+        :return: (pandas.DataFrame) -- dummy pg
+        """
+        return self.pg
+
+    def get_solar(self):
+        """Get solar.
+        :return: (pandas.DataFrame) -- dummy solar
+        """
+        return self.solar
+
+    def get_wind(self):
+        """Get wind.
+        :return: (pandas.DataFrame) -- dummy wind
+        """
+        return self.wind
+
+    def get_hydro(self):
+        """Get hydro.
+        :return: (pandas.DataFrame) -- dummy hydro
+        """
+        return self.hydro
+
+    @property
+    def __class__(self):
+        """If anyone asks, I'm an Analyze object!"""
+        return Analyze

powersimdata/tests/mock_scenario.py

@@ -0,0 +1,18 @@
+from powersimdata.scenario.scenario import Scenario
+from powersimdata.tests.mock_analyze import MockAnalyze
+
+
+class MockScenario:
+    def __init__(self, grid_attrs, **kwargs):
+        """Constructor.
+
+        :param dict grid_attrs: fields to be added to grid.
+        :param pandas.DataFrame pg: dummy pg
+        """
+        self.state = MockAnalyze(grid_attrs, **kwargs)
+        self.info = {id: '111'}
+
+    @property
+    def __class__(self):
+        """If anyone asks, I'm a Scenario object!"""
+        return Scenario

powersimdata/tests/test_mocks.py

@@ -0,0 +1,100 @@
+import unittest
+
+import pandas as pd
+
+from powersimdata.tests.mock_scenario import MockScenario
+from powersimdata.tests.mock_grid import MockGrid
+
+
+# plant_id is the index
+mock_plant = {
+    'plant_id': [101, 102, 103, 104, 105, 106],
+    'bus_id': [1001, 1002, 1003, 1004, 1005, 1006],
+    'type': ['solar', 'wind', 'ng', 'coal', 'dfo', 'hydro'],
+    'zone_id': [1, 2, 3, 1, 3, 2],
+    'GenFuelCost': [0, 0, 3.3, 4.4, 5.5, 0],
+    'Pmin': [0, 0, 0, 0, 0, 0],
+    'Pmax': [40, 80, 50, 150, 80, 60],
+}
+
+
+class TestMocks(unittest.TestCase):
+
+    def setUp(self):
+        self.period_num = 4
+        self.mock_pg = pd.DataFrame({
+            plant_id: [(i+1)*p for p in range(self.period_num)]
+            for i, plant_id in enumerate(mock_plant['plant_id'])})
+        self.mock_pg.set_index(pd.date_range(
+            start='2016-01-01', periods=self.period_num, freq='H'),
+            inplace=True)
+        self.mock_pg.index.name = 'UTC'
+
+        solar_plant_id = [plant_id
+                          for i, plant_id in enumerate(mock_plant['plant_id'])
+                          if mock_plant['type'][i] == 'solar']
+        self.mock_solar = self.mock_pg[solar_plant_id] * 2
+
+        wind_plant_id = [plant_id
+                         for i, plant_id in enumerate(mock_plant['plant_id'])
+                         if mock_plant['type'][i] == 'wind']
+        self.mock_wind = self.mock_pg[wind_plant_id] * 4
+
+        hydro_plant_id = [plant_id
+                          for i, plant_id in enumerate(mock_plant['plant_id'])
+                          if mock_plant['type'][i] == 'hydro']
+        self.mock_hydro = self.mock_pg[hydro_plant_id] * 1.5
+
+    # check that MockGrid is working correctly
+    def test_mock_grid_successes(self):
+        grid = MockGrid(grid_attrs={'plant': mock_plant})
+        self.assertTrue(isinstance(grid, object),
+                        'MockGrid should return an object')
+        self.assertTrue(hasattr(grid, 'plant'),
+                        'Plant property should be in the MockGrid')
+        self.assertEqual(len(grid.branch), 0, 
+                         'Branch dataframe should be empty in the MockGrid')
+
+    def test_mock_grid_failures(self):
+        with self.assertRaises(TypeError):
+            MockGrid(grid_attrs='foo')
+        with self.assertRaises(TypeError):
+            MockGrid(grid_attrs={1: 'foo'})
+        with self.assertRaises(ValueError):
+            MockGrid(grid_attrs={'foo': 'bar'})
+
+    def test_mock_pg_stored_properly(self):
+        scenario = MockScenario(
+            grid_attrs={'plant': mock_plant},
+            pg=self.mock_pg)
+        pg = scenario.state.get_pg()
+        err_msg = 'pg should have dimension (periodNum * len(plant))'
+        self.assertEqual(pg.shape, self.mock_pg.shape, err_msg)
+
+    def test_mock_solar_stored_properly(self):
+        scenario = MockScenario(
+            grid_attrs={'plant': mock_plant},
+            solar=self.mock_solar)
+        solar = scenario.state.get_solar()
+        err_msg = 'solar should have dimension (periodNum * len(solar_plant))'
+        self.assertEqual(solar.shape, self.mock_solar.shape, err_msg)
+
+    def test_mock_wind_stored_properly(self):
+        scenario = MockScenario(
+            grid_attrs={'plant': mock_plant},
+            wind=self.mock_wind)
+        wind = scenario.state.get_wind()
+        err_msg = 'wind should have dimension (periodNum * len(wind_plant))'
+        self.assertEqual(wind.shape, self.mock_wind.shape, err_msg)
+
+    def test_mock_hydro_stored_properly(self):
+        scenario = MockScenario(
+            grid_attrs={'plant': mock_plant},
+            hydro=self.mock_hydro)
+        hydro = scenario.state.get_hydro()
+        err_msg = 'hydro should have dimension (periodNum * len(hydro_plant))'
+        self.assertEqual(hydro.shape, self.mock_hydro.shape, err_msg)
+
+
+if __name__ == '__main__':
+    unittest.main()

powersimdata/utility/__init__.py

@@ -1 +1 @@
-__all__ = ['const.py', 'transfer_data']
+__all__ = ['const.py', 'constants', 'distance', 'transfer_data']