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

Make assert_equals work with np.ndarray

Browse Source
This commit is contained in:
Alinson S. Xavier
2021-05-20 10:41:38 -05:00
parent 310394b397
commit cdd38cdfb8
2 changed files with 20 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
tests/test_features.py
View File

@@ -13,6 +13,7 @@ from miplearn.solvers.tests import (
assert_equals,
_round,
)
import numpy as np
inf = float("inf")
@@ -134,3 +135,16 @@ def test_constraint_getindex() -> None:
],
),
)
def test_assert_equals() -> None:
assert_equals("hello", "hello")
assert_equals([1.0, 2.0], [1.0, 2.0])
assert_equals(
np.array([1.0, 2.0]),
np.array([1.0, 2.0]),
)
assert_equals(
np.array([[1.0, 2.0], [3.0, 4.0]]),
np.array([[1.0, 2.0], [3.0, 4.0]]),
)