ltbams 1.0.12__py3-none-any.whl → 1.0.13__py3-none-any.whl

tests/test_model.py

@@ -1,54 +0,0 @@
-import unittest
-import numpy as np
-
-import ams
-
-
-class TestModelMethods(unittest.TestCase):
-    """
-    Test methods of Model.
-    """
-
-    def setUp(self):
-        self.ss = ams.run(
-            ams.get_case("ieee14/ieee14.json"),
-            default_config=True,
-            no_output=True,
-        )
-
-    def test_model_set(self):
-        """
-        Test `Model.set()` method.
-        """
-
-        # set a single value
-        self.ss.PQ.set("p0", "PQ_1", "v", 0.25)
-        self.assertEqual(self.ss.PQ.p0.v[0], 0.25)
-
-        # set a list of values
-        self.ss.PQ.set("p0", ["PQ_1", "PQ_2"], "v", [0.26, 0.51])
-        np.testing.assert_equal(self.ss.PQ.p0.v[[0, 1]], [0.26, 0.51])
-
-        # set a list of values
-        self.ss.PQ.set("p0", ["PQ_3", "PQ_5"], "v", [0.52, 0.16])
-        np.testing.assert_equal(self.ss.PQ.p0.v[[2, 4]], [0.52, 0.16])
-
-        # set a list of idxes with a single element to an array of values
-        self.ss.PQ.set("p0", ["PQ_4"], "v", np.array([0.086]))
-        np.testing.assert_equal(self.ss.PQ.p0.v[3], 0.086)
-
-        # set an array of idxes with a single element to an array of values
-        self.ss.PQ.set("p0", np.array(["PQ_4"]), "v", np.array([0.096]))
-        np.testing.assert_equal(self.ss.PQ.p0.v[3], 0.096)
-
-        # set an array of idxes with a list of single value
-        self.ss.PQ.set("p0", np.array(["PQ_4"]), "v", 0.097)
-        np.testing.assert_equal(self.ss.PQ.p0.v[3], 0.097)
-
-    def test_model_repr(self):
-        """
-        Test `Model.__repr__()` method.
-        """
-
-        for mdl in self.ss.models.items():
-            print(mdl.__repr__())

tests/test_omodel.py

@@ -1,119 +0,0 @@
-import unittest
-
-import ams
-
-
-class TestOModelWrappers(unittest.TestCase):
-    """
-    Test wrappers in module `omodel`.
-    """
-
-    def setUp(self) -> None:
-        self.ss = ams.load(ams.get_case("matpower/case5.m"),
-                           setup=True,
-                           default_config=True,
-                           no_output=True,
-                           )
-
-    def test_ensure_symbols(self):
-        """
-        Test `ensure_symbols` wrapper.
-        """
-        self.assertFalse(self.ss.DCOPF._syms, "Symbols should not be ready before generation!")
-        self.ss.DCOPF.pd.parse()
-        self.assertTrue(self.ss.DCOPF._syms, "Symbols should be ready after generation!")
-
-    def test_ensure_mats_and_parsed(self):
-        """
-        Test `ensure_mats_and_parsed` wrapper.
-        """
-        self.assertFalse(self.ss.mats.initialized, "MatProcessor should not be initialized before initialization!")
-        self.assertFalse(self.ss.DCOPF.om.parsed, "OModel should be parsed after parsing!")
-        self.ss.DCOPF.pd.evaluate()
-        self.assertTrue(self.ss.mats.initialized, "MatProcessor should be initialized after parsing!")
-        self.assertTrue(self.ss.DCOPF.om.parsed, "OModle should be parsed after parsing!")
-
-
-class TestOModel(unittest.TestCase):
-    """
-    Test class `OModel`.
-    """
-
-    def setUp(self) -> None:
-        self.ss = ams.load(ams.get_case("matpower/case5.m"),
-                           setup=True,
-                           default_config=True,
-                           no_output=True,
-                           )
-
-    def test_initialized(self):
-        """
-        Test `OModel` initialization.
-        """
-        self.assertFalse(self.ss.DCOPF.om.initialized, "OModel shoud not be initialized before initialziation!")
-        self.ss.DCOPF.om.init()
-        self.assertTrue(self.ss.DCOPF.om.initialized, "OModel should be initialized!")
-
-    def test_uninitialized_after_nonparametric_update(self):
-        """
-        Test `OModel` initialization after nonparametric update.
-        """
-        self.ss.DCOPF.om.init()
-        self.assertTrue(self.ss.DCOPF.om.initialized, "OModel should be initialized after initialization!")
-        self.ss.DCOPF.update('ug')
-        self.assertTrue(self.ss.DCOPF.om.initialized, "OModel should be initialized after nonparametric update!")
-
-
-class TestOModelrepr(unittest.TestCase):
-    """
-    Test __repr__ in module `omodel`.
-    """
-
-    def setUp(self) -> None:
-        self.ss = ams.load(ams.get_case("5bus/pjm5bus_demo.json"),
-                           setup=True,
-                           default_config=True,
-                           no_output=True,
-                           )
-
-    def test_dcopf(self):
-        """
-        Test `DCOPF.om` __repr__().
-        """
-        self.ss.DCOPF.om.parse()
-        self.assertIsInstance(self.ss.DCOPF.pg.__repr__(), str)
-        self.assertIsInstance(self.ss.DCOPF.pmax.__repr__(), str)
-        self.assertIsInstance(self.ss.DCOPF.pb.__repr__(), str)
-        self.assertIsInstance(self.ss.DCOPF.plflb.__repr__(), str)
-        self.assertIsInstance(self.ss.DCOPF.obj.__repr__(), str)
-
-    def test_rted(self):
-        """
-        Test `RTED.om` __repr__().
-        """
-        self.ss.RTED.om.parse()
-        self.assertIsInstance(self.ss.RTED.pru.__repr__(), str)
-        self.assertIsInstance(self.ss.RTED.R10.__repr__(), str)
-        self.assertIsInstance(self.ss.RTED.pru.__repr__(), str)
-        self.assertIsInstance(self.ss.RTED.rru.__repr__(), str)
-        self.assertIsInstance(self.ss.RTED.obj.__repr__(), str)
-
-    def test_ed(self):
-        """
-        Test `ED.om` __repr__().
-        """
-        self.ss.ED.om.parse()
-        self.assertIsInstance(self.ss.ED.prs.__repr__(), str)
-        self.assertIsInstance(self.ss.ED.R30.__repr__(), str)
-        self.assertIsInstance(self.ss.ED.prsb.__repr__(), str)
-        self.assertIsInstance(self.ss.ED.obj.__repr__(), str)
-
-    def test_uc(self):
-        """
-        Test `UC.om` __repr__().
-        """
-        self.ss.UC.om.parse()
-        self.assertIsInstance(self.ss.UC.prns.__repr__(), str)
-        self.assertIsInstance(self.ss.UC.cdp.__repr__(), str)
-        self.assertIsInstance(self.ss.UC.prnsb.__repr__(), str)
-        self.assertIsInstance(self.ss.UC.obj.__repr__(), str)

tests/test_paths.py

@@ -1,89 +0,0 @@
-import unittest
-
-import os
-
-import ams
-from ams.utils.paths import list_cases, get_export_path
-
-
-class TestPaths(unittest.TestCase):
-    def setUp(self) -> None:
-        self.npcc = 'npcc/'
-        self.matpower = 'matpower/'
-        self.ieee14 = ams.get_case("ieee14/ieee14.raw")
-
-    def test_tree(self):
-        list_cases(self.npcc, no_print=True)
-        list_cases(self.matpower, no_print=True)
-
-    def test_relative_path(self):
-        ss = ams.run('ieee14.raw',
-                     input_path=ams.get_case('ieee14/', check=False),
-                     no_output=True, default_config=True,
-                     )
-        self.assertNotEqual(ss, None)
-
-
-class TestExportPath(unittest.TestCase):
-
-    def setUp(self) -> None:
-        self.ss = ams.load(ams.get_case('5bus/pjm5bus_demo.json'),
-                           setup=True, no_output=True)
-
-    def test_no_fullname(self):
-        """
-        Test export path with no full name specified.
-        """
-        self.ss.files.full_name = None
-
-        path, file_name = get_export_path(self.ss,
-                                          'DCOPF',
-                                          path=None,
-                                          fmt='csv')
-
-        dir_path, only_file_name = os.path.split(path)
-        self.assertTrue(os.path.exists(dir_path))
-        self.assertIsNotNone(only_file_name)
-        self.assertEqual(only_file_name, file_name)
-
-    def test_no_path(self):
-        """
-        Test export path with no path specified.
-        """
-        path, file_name = get_export_path(self.ss,
-                                          'DCOPF',
-                                          path=None,
-                                          fmt='csv')
-
-        dir_path, only_file_name = os.path.split(path)
-        self.assertTrue(os.path.exists(dir_path))
-        self.assertIsNotNone(only_file_name)
-        self.assertEqual(only_file_name, file_name)
-
-    def test_current_path(self):
-        """
-        Test export path with current path specified.
-        """
-        path, file_name = get_export_path(self.ss,
-                                          'DCOPF',
-                                          path='.',
-                                          fmt='csv')
-
-        dir_path, only_file_name = os.path.split(path)
-        self.assertTrue(os.path.exists(dir_path))
-        self.assertIsNotNone(only_file_name)
-        self.assertEqual(only_file_name, file_name)
-
-    def test_path_with_file_name(self):
-        """
-        Test export path with path and file name specified.
-        """
-        path, file_name = get_export_path(self.ss,
-                                          'DCOPF',
-                                          path='./test_export.csv',
-                                          fmt='csv',)
-
-        dir_path, only_file_name = os.path.split(path)
-        self.assertTrue(os.path.exists(dir_path))
-        self.assertEqual(only_file_name, 'test_export.csv')
-        self.assertEqual(only_file_name, file_name)

tests/test_report.py

@@ -1,251 +0,0 @@
-"""
-Test system report.
-"""
-import unittest
-import os
-
-import ams
-from ams.shared import skip_unittest_without_MISOCP
-
-
-import logging
-
-logger = logging.getLogger(__name__)
-
-
-class TestReport(unittest.TestCase):
-    """
-    Tests for Report class.
-    """
-
-    def setUp(self) -> None:
-        self.ss = ams.main.load(
-            ams.get_case("5bus/pjm5bus_demo.json"),
-            default_config=True,
-            no_output=True,
-        )
-        self.expected_report = 'pjm5bus_demo_out.txt'
-
-    def test_no_output(self):
-        """
-        Test no output.
-        """
-        self.assertTrue(self.ss.files.no_output)
-        self.assertFalse(self.ss.report())
-
-    def test_no_report(self):
-        """
-        Test report with no solved routine.
-        """
-        self.ss.files.no_output = False
-        self.assertTrue(self.ss.report())
-        self.assertTrue(os.path.exists(self.expected_report))
-
-        with open(self.expected_report, "r") as report_file:
-            file_contents = report_file.read()
-            self.assertNotIn("DCOPF", file_contents)
-
-        os.remove(self.expected_report)
-
-    def test_DCOPF_report(self):
-        """
-        Test report with DCOPF solved.
-        """
-        self.ss.files.no_output = False
-        self.ss.DCOPF.run(solver='CLARABEL')
-        self.assertTrue(self.ss.report())
-        self.assertTrue(os.path.exists(self.expected_report))
-
-        with open(self.expected_report, "r") as report_file:
-            file_contents = report_file.read()
-            self.assertIn("DCOPF", file_contents)
-        os.remove(self.expected_report)
-
-        self.ss.DCOPF.export_csv()
-        self.assertTrue(os.path.exists('pjm5bus_demo_DCOPF.csv'))
-        os.remove('pjm5bus_demo_DCOPF.csv')
-
-    def test_DCPF_report(self):
-        """
-        Test report with DCPF solved.
-        """
-        self.ss.files.no_output = False
-        self.ss.DCPF.run(solver='CLARABEL')
-        self.assertTrue(self.ss.report())
-        self.assertTrue(os.path.exists(self.expected_report))
-
-        with open(self.expected_report, "r") as report_file:
-            file_contents = report_file.read()
-            self.assertIn("DCPF", file_contents)
-        os.remove(self.expected_report)
-
-        self.ss.DCPF.export_csv()
-        self.assertTrue(os.path.exists('pjm5bus_demo_DCPF.csv'))
-        os.remove('pjm5bus_demo_DCPF.csv')
-
-    def test_RTED_report(self):
-        """
-        Test report with RTED solved.
-        """
-        self.ss.files.no_output = False
-        self.ss.RTED.run(solver='CLARABEL')
-        self.assertTrue(self.ss.report())
-        self.assertTrue(os.path.exists(self.expected_report))
-
-        with open(self.expected_report, "r") as report_file:
-            file_contents = report_file.read()
-            self.assertIn("RTED", file_contents)
-        os.remove(self.expected_report)
-
-        self.ss.RTED.export_csv()
-        self.assertTrue(os.path.exists('pjm5bus_demo_RTED.csv'))
-        os.remove('pjm5bus_demo_RTED.csv')
-
-    def test_RTEDDG_report(self):
-        """
-        Test report with RTEDDG solved.
-        """
-        self.ss.files.no_output = False
-        self.ss.RTEDDG.run(solver='CLARABEL')
-        self.assertTrue(self.ss.report())
-        self.assertTrue(os.path.exists(self.expected_report))
-
-        with open(self.expected_report, "r") as report_file:
-            file_contents = report_file.read()
-            self.assertIn("RTEDDG", file_contents)
-        os.remove(self.expected_report)
-
-        self.ss.RTEDDG.export_csv()
-        self.assertTrue(os.path.exists('pjm5bus_demo_RTEDDG.csv'))
-        os.remove('pjm5bus_demo_RTEDDG.csv')
-
-    @skip_unittest_without_MISOCP
-    def test_RTEDES_report(self):
-        """
-        Test report with RTEDES solved.
-        """
-        self.ss.files.no_output = False
-        self.ss.RTEDES.run(solver='SCIP')
-        self.assertTrue(self.ss.report())
-        self.assertTrue(os.path.exists(self.expected_report))
-
-        with open(self.expected_report, "r") as report_file:
-            file_contents = report_file.read()
-            self.assertIn("RTEDES", file_contents)
-        os.remove(self.expected_report)
-
-        self.ss.RTEDES.export_csv()
-        self.assertTrue(os.path.exists('pjm5bus_demo_RTEDES.csv'))
-        os.remove('pjm5bus_demo_RTEDES.csv')
-
-    def test_ED_report(self):
-        """
-        Test report with ED solved.
-        """
-        self.ss.files.no_output = False
-        self.ss.ED.run(solver='CLARABEL')
-        self.assertTrue(self.ss.report())
-        self.assertTrue(os.path.exists(self.expected_report))
-
-        with open(self.expected_report, "r") as report_file:
-            file_contents = report_file.read()
-            self.assertIn("ED", file_contents)
-        os.remove(self.expected_report)
-
-        self.ss.ED.export_csv()
-        self.assertTrue(os.path.exists('pjm5bus_demo_ED.csv'))
-        os.remove('pjm5bus_demo_ED.csv')
-
-    def test_EDDG_report(self):
-        """
-        Test report with EDDG solved.
-        """
-        self.ss.files.no_output = False
-        self.ss.EDDG.run(solver='CLARABEL')
-        self.assertTrue(self.ss.report())
-        self.assertTrue(os.path.exists(self.expected_report))
-
-        with open(self.expected_report, "r") as report_file:
-            file_contents = report_file.read()
-            self.assertIn("EDDG", file_contents)
-        os.remove(self.expected_report)
-
-        self.ss.EDDG.export_csv()
-        self.assertTrue(os.path.exists('pjm5bus_demo_EDDG.csv'))
-        os.remove('pjm5bus_demo_EDDG.csv')
-
-    @skip_unittest_without_MISOCP
-    def test_EDES_report(self):
-        """
-        Test report with EDES solved.
-        """
-        self.ss.files.no_output = False
-        self.ss.EDES.run(solver='SCIP')
-        self.assertTrue(self.ss.report())
-        self.assertTrue(os.path.exists(self.expected_report))
-
-        with open(self.expected_report, "r") as report_file:
-            file_contents = report_file.read()
-            self.assertIn("EDES", file_contents)
-        os.remove(self.expected_report)
-
-        self.ss.EDES.export_csv()
-        self.assertTrue(os.path.exists('pjm5bus_demo_EDES.csv'))
-        os.remove('pjm5bus_demo_EDES.csv')
-
-    @skip_unittest_without_MISOCP
-    def test_UC_report(self):
-        """
-        Test report with UC solved.
-        """
-        self.ss.files.no_output = False
-        self.ss.UC.run(solver='SCIP')
-        self.assertTrue(self.ss.report())
-        self.assertTrue(os.path.exists(self.expected_report))
-
-        with open(self.expected_report, "r") as report_file:
-            file_contents = report_file.read()
-            self.assertIn("UC", file_contents)
-        os.remove(self.expected_report)
-
-        self.ss.UC.export_csv()
-        self.assertTrue(os.path.exists('pjm5bus_demo_UC.csv'))
-        os.remove('pjm5bus_demo_UC.csv')
-
-    @skip_unittest_without_MISOCP
-    def test_UCDG_report(self):
-        """
-        Test report with UCDG solved.
-        """
-        self.ss.files.no_output = False
-        self.ss.UCDG.run(solver='SCIP')
-        self.assertTrue(self.ss.report())
-        self.assertTrue(os.path.exists(self.expected_report))
-
-        with open(self.expected_report, "r") as report_file:
-            file_contents = report_file.read()
-            self.assertIn("UCDG", file_contents)
-        os.remove(self.expected_report)
-
-        self.ss.UCDG.export_csv()
-        self.assertTrue(os.path.exists('pjm5bus_demo_UCDG.csv'))
-        os.remove('pjm5bus_demo_UCDG.csv')
-
-    @skip_unittest_without_MISOCP
-    def test_UCES_report(self):
-        """
-        Test report with UCES solved.
-        """
-        self.ss.files.no_output = False
-        self.ss.UCES.run(solver='SCIP')
-        self.assertTrue(self.ss.report())
-        self.assertTrue(os.path.exists(self.expected_report))
-
-        with open(self.expected_report, "r") as report_file:
-            file_contents = report_file.read()
-            self.assertIn("UCES", file_contents)
-        os.remove(self.expected_report)
-
-        self.ss.UCES.export_csv()
-        self.assertTrue(os.path.exists('pjm5bus_demo_UCES.csv'))
-        os.remove('pjm5bus_demo_UCES.csv')

tests/test_repr.py

@@ -1,21 +0,0 @@
-import contextlib
-import unittest
-
-import ams
-
-
-class TestRepr(unittest.TestCase):
-    """Test __repr__"""
-    def setUp(self):
-        self.ss = ams.run(ams.get_case("ieee39/ieee39_uced.xlsx"),
-                          no_output=True,
-                          default_config=True,
-                          )
-
-    def test_print_model_repr(self):
-        """
-        Print out Model ``cache``'s fields and values.
-        """
-        with contextlib.redirect_stdout(None):
-            for model in self.ss.models.values():
-                print(model.cache.__dict__)