Skip to content

Commit

Permalink
register test_sparse in run_tests.py
Browse files Browse the repository at this point in the history
  • Loading branch information
frrad committed Sep 2, 2024
1 parent 0bdbbd5 commit bc0d6a1
Show file tree
Hide file tree
Showing 2 changed files with 28 additions and 24 deletions.
22 changes: 11 additions & 11 deletions pulp/tests/run_tests.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
import unittest
import pulp
from pulp.tests import test_pulp, test_examples, test_gurobipy_env
from pulp.tests import test_pulp, test_examples, test_gurobipy_env, test_sparse


def pulpTestAll(test_docs=False):
Expand All @@ -16,20 +16,20 @@ def pulpTestAll(test_docs=False):
raise pulp.PulpError("Tests Failed")


def get_test_suite(test_docs=False):
# Tests
def get_test_suite(test_docs: bool = False) -> unittest.TestSuite:
loader = unittest.TestLoader()
suite_all = unittest.TestSuite()
# we get suite with all PuLP tests
pulp_solver_tests = loader.loadTestsFromModule(test_pulp)
suite_all.addTests(pulp_solver_tests)
# Add tests for gurobipy env
gurobipy_env = loader.loadTestsFromModule(test_gurobipy_env)
suite_all.addTests(gurobipy_env)

suite_all.addTests(loader.loadTestsFromModule(test_pulp))
suite_all.addTests(loader.loadTestsFromModule(test_sparse))
suite_all.addTests(loader.loadTestsFromModule(test_gurobipy_env))

# We add examples and docs tests
if test_docs:
docs_examples = loader.loadTestsFromTestCase(test_examples.Examples_DocsTests)
suite_all.addTests(docs_examples)
suite_all.addTests(
loader.loadTestsFromTestCase(test_examples.Examples_DocsTests)
)

return suite_all


Expand Down
30 changes: 17 additions & 13 deletions pulp/tests/test_sparse.py
Original file line number Diff line number Diff line change
@@ -1,16 +1,20 @@
import unittest

from pulp.sparse import Matrix


def test_sparse():
rows = list(range(10))
cols = list(range(50, 60))
mat = Matrix(rows, cols)
mat.add(1, 52, "item")
mat.add(2, 54, "stuff")
assert mat.col_based_arrays() == (
2,
[0, 0, 0, 1, 1, 2, 2, 2, 2, 2, 2],
[0, 0, 1, 0, 1, 0, 0, 0, 0, 0],
[1, 2],
["item", "stuff"],
)
class SparseTest(unittest.TestCase):
def test_sparse(self):
assert False
rows = list(range(10))
cols = list(range(50, 60))
mat = Matrix(rows, cols)
mat.add(1, 52, "item")
mat.add(2, 54, "stuff")
assert mat.col_based_arrays() == (
2,
[0, 0, 0, 1, 1, 2, 2, 2, 2, 2, 2],
[0, 0, 1, 0, 1, 0, 0, 0, 0, 0],
[1, 2],
["item", "stuff"],
)

0 comments on commit bc0d6a1

Please sign in to comment.