linopy/test/test_dtypes.py at 2f3e87eac6ccdfca5fe9d9bf0a5b433b10f65bea · PyPSA/linopy · GitHub

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
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
"""Tests for int32 default label dtype."""

import numpy as np
import pytest

from linopy import Model
from linopy.constants import DEFAULT_LABEL_DTYPE


def test_default_label_dtype_is_int32():
    assert DEFAULT_LABEL_DTYPE == np.int32


def test_variable_labels_are_int32():
    m = Model()
    x = m.add_variables(lower=0, upper=10, coords=[range(5)], name="x")
    assert x.labels.dtype == np.int32


def test_constraint_labels_are_int32():
    m = Model()
    x = m.add_variables(lower=0, upper=10, coords=[range(5)], name="x")
    m.add_constraints(x >= 1, name="c")
    assert m.constraints["c"].labels.dtype == np.int32


def test_expression_vars_are_int32():
    m = Model()
    x = m.add_variables(lower=0, upper=10, coords=[range(5)], name="x")
    expr = 2 * x + 1
    assert expr.vars.dtype == np.int32


def test_solve_with_int32_labels():
    m = Model()
    x = m.add_variables(lower=0, upper=10, name="x")
    y = m.add_variables(lower=0, upper=10, name="y")
    m.add_constraints(x + y <= 15, name="c1")
    m.add_objective(x + 2 * y, sense="max")
    m.solve("highs")
    assert m.objective.value == pytest.approx(25.0)


def test_overflow_guard_variables():
    m = Model()
    m._xCounter = np.iinfo(np.int32).max - 1
    with pytest.raises(ValueError, match="exceeds the maximum"):
        m.add_variables(lower=0, upper=1, coords=[range(5)], name="x")


def test_overflow_guard_constraints():
    m = Model()
    x = m.add_variables(lower=0, upper=1, coords=[range(5)], name="x")
    m._cCounter = np.iinfo(np.int32).max - 1
    with pytest.raises(ValueError, match="exceeds the maximum"):
        m.add_constraints(x >= 0, name="c")