diff --git a/tests/st/ops/ascend/test_bijector/test_exp.py b/tests/st/probability/bijector/test_exp.py
similarity index 100%
rename from tests/st/ops/ascend/test_bijector/test_exp.py
rename to tests/st/probability/bijector/test_exp.py
diff --git a/tests/st/ops/ascend/test_bijector/test_power_transform.py b/tests/st/probability/bijector/test_power_transform.py
similarity index 100%
rename from tests/st/ops/ascend/test_bijector/test_power_transform.py
rename to tests/st/probability/bijector/test_power_transform.py
diff --git a/tests/st/ops/ascend/test_bijector/test_scalar_affine.py b/tests/st/probability/bijector/test_scalar_affine.py
similarity index 100%
rename from tests/st/ops/ascend/test_bijector/test_scalar_affine.py
rename to tests/st/probability/bijector/test_scalar_affine.py
diff --git a/tests/st/ops/ascend/test_bijector/test_softplus.py b/tests/st/probability/bijector/test_softplus.py
similarity index 100%
rename from tests/st/ops/ascend/test_bijector/test_softplus.py
rename to tests/st/probability/bijector/test_softplus.py
diff --git a/tests/st/ops/ascend/test_distribution/test_bernoulli.py b/tests/st/probability/distribution/test_bernoulli.py
similarity index 100%
rename from tests/st/ops/ascend/test_distribution/test_bernoulli.py
rename to tests/st/probability/distribution/test_bernoulli.py
diff --git a/tests/st/ops/ascend/test_distribution/test_exponential.py b/tests/st/probability/distribution/test_exponential.py
similarity index 100%
rename from tests/st/ops/ascend/test_distribution/test_exponential.py
rename to tests/st/probability/distribution/test_exponential.py
diff --git a/tests/st/ops/ascend/test_distribution/test_geometric.py b/tests/st/probability/distribution/test_geometric.py
similarity index 100%
rename from tests/st/ops/ascend/test_distribution/test_geometric.py
rename to tests/st/probability/distribution/test_geometric.py
diff --git a/tests/st/ops/ascend/test_distribution/test_normal.py b/tests/st/probability/distribution/test_normal.py
similarity index 100%
rename from tests/st/ops/ascend/test_distribution/test_normal.py
rename to tests/st/probability/distribution/test_normal.py
diff --git a/tests/st/ops/ascend/test_distribution/test_uniform.py b/tests/st/probability/distribution/test_uniform.py
similarity index 100%
rename from tests/st/ops/ascend/test_distribution/test_uniform.py
rename to tests/st/probability/distribution/test_uniform.py
diff --git a/tests/ut/python/nn/bijector/test_exp.py b/tests/ut/python/nn/probability/bijector/test_exp.py
similarity index 100%
rename from tests/ut/python/nn/bijector/test_exp.py
rename to tests/ut/python/nn/probability/bijector/test_exp.py
diff --git a/tests/ut/python/nn/bijector/test_power_transform.py b/tests/ut/python/nn/probability/bijector/test_power_transform.py
similarity index 100%
rename from tests/ut/python/nn/bijector/test_power_transform.py
rename to tests/ut/python/nn/probability/bijector/test_power_transform.py
diff --git a/tests/ut/python/nn/bijector/test_scalar_affine.py b/tests/ut/python/nn/probability/bijector/test_scalar_affine.py
similarity index 100%
rename from tests/ut/python/nn/bijector/test_scalar_affine.py
rename to tests/ut/python/nn/probability/bijector/test_scalar_affine.py
diff --git a/tests/ut/python/nn/bijector/test_softplus.py b/tests/ut/python/nn/probability/bijector/test_softplus.py
similarity index 100%
rename from tests/ut/python/nn/bijector/test_softplus.py
rename to tests/ut/python/nn/probability/bijector/test_softplus.py
diff --git a/tests/ut/python/nn/distribution/test_bernoulli.py b/tests/ut/python/nn/probability/distribution/test_bernoulli.py
similarity index 100%
rename from tests/ut/python/nn/distribution/test_bernoulli.py
rename to tests/ut/python/nn/probability/distribution/test_bernoulli.py
diff --git a/tests/ut/python/nn/distribution/test_exponential.py b/tests/ut/python/nn/probability/distribution/test_exponential.py
similarity index 100%
rename from tests/ut/python/nn/distribution/test_exponential.py
rename to tests/ut/python/nn/probability/distribution/test_exponential.py
diff --git a/tests/ut/python/nn/distribution/test_geometric.py b/tests/ut/python/nn/probability/distribution/test_geometric.py
similarity index 100%
rename from tests/ut/python/nn/distribution/test_geometric.py
rename to tests/ut/python/nn/probability/distribution/test_geometric.py
diff --git a/tests/ut/python/nn/distribution/test_normal.py b/tests/ut/python/nn/probability/distribution/test_normal.py
similarity index 100%
rename from tests/ut/python/nn/distribution/test_normal.py
rename to tests/ut/python/nn/probability/distribution/test_normal.py
diff --git a/tests/ut/python/nn/distribution/test_uniform.py b/tests/ut/python/nn/probability/distribution/test_uniform.py
similarity index 100%
rename from tests/ut/python/nn/distribution/test_uniform.py
rename to tests/ut/python/nn/probability/distribution/test_uniform.py
diff --git a/tests/ut/python/nn/distribution/test_utils.py b/tests/ut/python/nn/probability/distribution/test_utils.py
similarity index 100%
rename from tests/ut/python/nn/distribution/test_utils.py
rename to tests/ut/python/nn/probability/distribution/test_utils.py