ANL-CEEESA/MIPLearn
1
0
Fork 0
mirror of https://github.com/ANL-CEEESA/MIPLearn.git synced 2025-12-06 09:28:51 -06:00
Code Issues Packages Projects Releases Wiki Activity

Fix failing tests

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-04-04 08:55:55 -05:00
parent 96e7a0946e
commit b4770c6c0a
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
tests/components/test_objective.py
View File

@@ -1,6 +1,7 @@
# MIPLearn: Extensible Framework for Learning-Enhanced Mixed-Integer Optimization # MIPLearn: Extensible Framework for Learning-Enhanced Mixed-Integer Optimization
# Copyright (C) 2020, UChicago Argonne, LLC. All rights reserved. # Copyright (C) 2020, UChicago Argonne, LLC. All rights reserved.
# Released under the modified BSD license. See COPYING.md for more details. # Released under the modified BSD license. See COPYING.md for more details.
from typing import Hashable, Dict
from unittest.mock import Mock from unittest.mock import Mock
import pytest import pytest
@@ -103,11 +104,11 @@ def test_sample_xy_without_ub(
def test_fit_xy() -> None: def test_fit_xy() -> None:
x = { x: Dict[Hashable, np.ndarray] = {
"Lower bound": np.array([[0.0, 0.0], [1.0, 2.0]]), "Lower bound": np.array([[0.0, 0.0], [1.0, 2.0]]),
"Upper bound": np.array([[0.0, 0.0], [1.0, 2.0]]), "Upper bound": np.array([[0.0, 0.0], [1.0, 2.0]]),
} }
y = { y: Dict[Hashable, np.ndarray] = {
"Lower bound": np.array([[100.0]]), "Lower bound": np.array([[100.0]]),
"Upper bound": np.array([[200.0]]), "Upper bound": np.array([[200.0]]),
} }
@@ -141,11 +142,11 @@ def test_fit_xy() -> None:
def test_fit_xy_without_ub() -> None: def test_fit_xy_without_ub() -> None:
x = { x: Dict[Hashable, np.ndarray] = {
"Lower bound": np.array([[0.0, 0.0], [1.0, 2.0]]), "Lower bound": np.array([[0.0, 0.0], [1.0, 2.0]]),
"Upper bound": np.array([[0.0, 0.0], [1.0, 2.0]]), "Upper bound": np.array([[0.0, 0.0], [1.0, 2.0]]),
} }
y = { y: Dict[Hashable, np.ndarray] = {
"Lower bound": np.array([[100.0]]), "Lower bound": np.array([[100.0]]),
} }
reg = Mock(spec=Regressor) reg = Mock(spec=Regressor)