From bc0d6a1df24dfca3eef369a18e86677c8e237a76 Mon Sep 17 00:00:00 2001 From: Frederick Robinson Date: Sun, 1 Sep 2024 20:41:02 -0700 Subject: [PATCH] register test_sparse in `run_tests.py` --- pulp/tests/run_tests.py | 22 +++++++++++----------- pulp/tests/test_sparse.py | 30 +++++++++++++++++------------- 2 files changed, 28 insertions(+), 24 deletions(-) diff --git a/pulp/tests/run_tests.py b/pulp/tests/run_tests.py index 0a6abd2a..151cb05e 100644 --- a/pulp/tests/run_tests.py +++ b/pulp/tests/run_tests.py @@ -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): @@ -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 diff --git a/pulp/tests/test_sparse.py b/pulp/tests/test_sparse.py index 45fefe5f..37fcfabf 100644 --- a/pulp/tests/test_sparse.py +++ b/pulp/tests/test_sparse.py @@ -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"], + )