refactor "test" to a test

pulp/sparse.py

@@ -76,13 +76,3 @@ def col_based_arrays(self):
             lenBase.append(len(elemBase) - startsBase[-1])
         startsBase.append(len(elemBase))
         return numEls, startsBase, lenBase, indBase, elemBase
-
-
-if __name__ == "__main__":
-    """unit test"""
-    rows = list(range(10))
-    cols = list(range(50, 60))
-    mat = Matrix(rows, cols)
-    mat.add(1, 52, "item")
-    mat.add(2, 54, "stuff")
-    print(mat.col_based_arrays())

pulp/tests/run_tests.py

@@ -1,6 +1,7 @@
 import unittest
+
 import pulp
-from pulp.tests import test_pulp, test_examples, test_gurobipy_env
+from pulp.tests import test_examples, test_gurobipy_env, test_pulp, test_sparse
 
 
 def pulpTestAll(test_docs=False):
@@ -16,20 +17,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
 
 

pulp/tests/test_sparse.py

@@ -0,0 +1,19 @@
+import unittest
+
+from pulp.sparse import Matrix
+
+
+class SparseTest(unittest.TestCase):
+    def test_sparse(self):
+        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"],
+        )