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/test_sparse.py

@@ -0,0 +1,16 @@
+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"],
+    )