gridoxide 0.1.0

AC power flow and state estimation solver with a sparse Newton-Raphson core
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30

"""The pglib-opf fixtures, converted once per session by
`tests/data/pglib_opf.py` from the `benchmark-grids` submodule."""

import atexit
import functools
import importlib.util
import shutil
import tempfile
from pathlib import Path

import pytest

_SCRIPT = Path(__file__).resolve().parents[2] / "tests" / "data" / "pglib_opf.py"
_spec = importlib.util.spec_from_file_location("pglib_opf", _SCRIPT)
pglib_opf = importlib.util.module_from_spec(_spec)
_spec.loader.exec_module(pglib_opf)

CASES = pglib_opf.CASES
PIECEWISE = pglib_opf.PIECEWISE


@functools.cache
def fixtures() -> Path:
    """Skips the calling module when the submodule or numpy is missing."""
    if not pglib_opf.SOURCE.is_dir():
        pytest.skip("benchmark-grids submodule not initialized", allow_module_level=True)
    pytest.importorskip("numpy")
    out = Path(tempfile.mkdtemp(prefix="pglib-opf-"))
    atexit.register(shutil.rmtree, out, ignore_errors=True)
    return pglib_opf.generate(out)