[Sparse] Namespace migration and doc update (#5088)

* update

* lint

* lint

* comments
Co-authored-by: Ubuntu <ubuntu@ip-172-31-36-188.ap-northeast-1.compute.internal>

python/dgl/mock_sparse2/unary_op_diag.py → python/dgl/sparse/unary_op_diag.py

 """DGL unary operators for diagonal matrix module."""
 # pylint: disable= invalid-name
-from .diag_matrix import DiagMatrix, diag
+from .diag_matrix import diag, DiagMatrix
 
 
 def neg(D: DiagMatrix) -> DiagMatrix:

tests/pytorch/mock_sparse2/__init__.py

-""" DGL mock_sparse2 tests"""
\ No newline at end of file

tests/pytorch/sparse/__init__.py

+""" DGL sparse tests"""

tests/pytorch/mock_sparse2/test_diag.py → tests/pytorch/sparse/test_diag.py

@@ -4,7 +4,7 @@ import backend as F
 import pytest
 import torch
 
-from dgl.mock_sparse2 import diag, DiagMatrix, identity
+from dgl.sparse import diag, DiagMatrix, identity
 
 # TODO(#4818): Skipping tests on win.
 if not sys.platform.startswith("linux"):

tests/pytorch/mock_sparse2/test_elementwise_op.py → tests/pytorch/sparse/test_elementwise_op.py

@@ -4,8 +4,13 @@ import backend as F
 import pytest
 import torch
 
-from dgl.mock_sparse2 import (add, create_from_coo, create_from_csc,
-                              create_from_csr, diag)
+from dgl.sparse import (
+    add,
+    create_from_coo,
+    create_from_csc,
+    create_from_csr,
+    diag,
+)
 
 # TODO(#4818): Skipping tests on win.
 if not sys.platform.startswith("linux"):

tests/pytorch/mock_sparse2/test_elementwise_op_diag.py → tests/pytorch/sparse/test_elementwise_op_diag.py

@@ -5,7 +5,7 @@ import backend as F
 import pytest
 import torch
 
-from dgl.mock_sparse2 import diag, power
+from dgl.sparse import diag, power
 
 # TODO(#4818): Skipping tests on win.
 if not sys.platform.startswith("linux"):
@@ -18,9 +18,7 @@ def all_close_sparse(A, B):
     assert A.shape == B.shape
 
 
-@pytest.mark.parametrize(
-    "op", [operator.sub, operator.mul, operator.truediv]
-)
+@pytest.mark.parametrize("op", [operator.sub, operator.mul, operator.truediv])
 def test_diag_op_diag(op):
     ctx = F.ctx()
     shape = (3, 4)

tests/pytorch/mock_sparse2/test_elementwise_op_sp.py → tests/pytorch/sparse/test_elementwise_op_sp.py

@@ -4,7 +4,7 @@ import backend as F
 import pytest
 import torch
 
-from dgl.mock_sparse2 import create_from_coo, power
+from dgl.sparse import create_from_coo, power
 
 # TODO(#4818): Skipping tests on win.
 if not sys.platform.startswith("linux"):

tests/pytorch/mock_sparse2/test_example.py → tests/pytorch/sparse/test_example.py

 import os
 import subprocess
 import sys
+
 import pytest
 
 # TODO(#4818): Skipping tests on win.

tests/pytorch/mock_sparse2/test_matmul.py → tests/pytorch/sparse/test_matmul.py

@@ -4,7 +4,7 @@ import backend as F
 import pytest
 import torch
 
-from dgl.mock_sparse2 import bspmm, create_from_coo, val_like
+from dgl.sparse import bspmm, create_from_coo, val_like
 
 from .utils import (
     clone_detach_and_grad,

tests/pytorch/mock_sparse2/test_reduction.py → tests/pytorch/sparse/test_reduction.py

@@ -4,7 +4,7 @@ import sys
 
 import backend as F
 
-import dgl.mock_sparse2 as dglsp
+import dgl.sparse as dglsp
 import pytest
 import torch
 

tests/pytorch/mock_sparse2/test_sddmm.py → tests/pytorch/sparse/test_sddmm.py

@@ -3,7 +3,7 @@ import sys
 import backend as F
 import pytest
 import torch
-from dgl.mock_sparse2 import bsddmm, sddmm
+from dgl.sparse import bsddmm, sddmm
 
 from .utils import clone_detach_and_grad, rand_coo, rand_csc, rand_csr
 

tests/pytorch/mock_sparse2/test_softmax.py → tests/pytorch/sparse/test_softmax.py

@@ -6,7 +6,7 @@ import dgl
 import pytest
 import torch
 
-from dgl.mock_sparse2 import create_from_coo, softmax
+from dgl.sparse import create_from_coo, softmax
 
 # TODO(#4818): Skipping tests on win.
 if not sys.platform.startswith("linux"):

tests/pytorch/mock_sparse2/test_sparse_matrix.py → tests/pytorch/sparse/test_sparse_matrix.py

@@ -4,7 +4,7 @@ import backend as F
 import pytest
 import torch
 
-from dgl.mock_sparse2 import (
+from dgl.sparse import (
     create_from_coo,
     create_from_csc,
     create_from_csr,

tests/pytorch/mock_sparse2/test_transpose.py → tests/pytorch/sparse/test_transpose.py

-import pytest
-import torch
 import sys
 
-from dgl.mock_sparse2 import diag, create_from_coo
-
 import backend as F
+import pytest
+import torch
+
+from dgl.sparse import create_from_coo, diag
 
 # TODO(#4818): Skipping tests on win.
 if not sys.platform.startswith("linux"):
     pytest.skip("skipping tests on win", allow_module_level=True)
 
+
 @pytest.mark.parametrize("val_shape", [(3,), (3, 2)])
 @pytest.mark.parametrize("mat_shape", [None, (3, 5), (5, 3)])
 def test_diag_matrix_transpose(val_shape, mat_shape):
@@ -43,4 +44,4 @@ def test_sparse_matrix_transpose(dense_dim, row, col, extra_shape):
     assert mat.shape == mat_shape[::-1]
     assert torch.allclose(mat_val, val)
     assert torch.allclose(mat_row, col)
-    assert torch.allclose(mat_col, row)
\ No newline at end of file
+    assert torch.allclose(mat_col, row)

tests/pytorch/mock_sparse2/test_unary_op_diag.py → tests/pytorch/sparse/test_unary_op_diag.py

-import pytest
-import torch
 import sys
 
 import backend as F
+import pytest
+import torch
 
-from dgl.mock_sparse2 import diag
+from dgl.sparse import diag
 
 
 def test_neg():

tests/pytorch/mock_sparse2/test_unary_op_sp.py → tests/pytorch/sparse/test_unary_op_sp.py

-import pytest
-import torch
 import sys
 
 import backend as F
+import pytest
+import torch
 
-from dgl.mock_sparse2 import create_from_coo
+from dgl.sparse import create_from_coo
 
 # TODO(#4818): Skipping tests on win.
 if not sys.platform.startswith("linux"):

tests/pytorch/mock_sparse2/utils.py → tests/pytorch/sparse/utils.py

 import numpy as np
 import torch
 
-from dgl.mock_sparse2 import (
+from dgl.sparse import (
     create_from_coo,
     create_from_csc,
     create_from_csr,