pymc-devs · ricardoV94 · Jul 19, 2024 · Jul 19, 2024 · Jul 19, 2024 · Jul 19, 2024
diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
@@ -7,6 +7,7 @@ on:
     paths:
       - ".github/workflows/*"
       - "pymc_experimental/**"
+      - "tests/**"
       - "setup.py"
       - "pyproject.toml"
       - "buildosx"
@@ -20,7 +21,7 @@ jobs:
         os: [ubuntu-latest]
         python-version: ["3.10"]
         test-subset:
-          - pymc_experimental/tests
+          - tests
       fail-fast: false
     runs-on: ${{ matrix.os }}
     env:
@@ -58,7 +59,7 @@ jobs:
         os: [windows-latest]
         python-version: ["3.12"]
         test-subset:
-          - pymc_experimental/tests
+          - tests
       fail-fast: false
     runs-on: ${{ matrix.os }}
     env:

diff --git a/codecov.yml b/codecov.yml
@@ -22,7 +22,7 @@ coverage:
         base: auto
 
 ignore:
-  - "pymc_experimental/tests/*"
+  - "tests/*"
 
 comment:
   layout: "reach, diff, flags, files"

diff --git a/pymc_experimental/tests/__init__.py → tests/__init__.py b/pymc_experimental/tests/__init__.py → tests/__init__.py
diff --git a/...erimental/tests/distributions/__init__.py → tests/distributions/__init__.py b/...erimental/tests/distributions/__init__.py → tests/distributions/__init__.py
diff --git a/...al/tests/distributions/test_continuous.py → tests/distributions/test_continuous.py b/...al/tests/distributions/test_continuous.py → tests/distributions/test_continuous.py
diff --git a/...ntal/tests/distributions/test_discrete.py → tests/distributions/test_discrete.py b/...ntal/tests/distributions/test_discrete.py → tests/distributions/test_discrete.py
diff --git a/...stributions/test_discrete_markov_chain.py → ...stributions/test_discrete_markov_chain.py b/...stributions/test_discrete_markov_chain.py → ...stributions/test_discrete_markov_chain.py
diff --git a/.../tests/distributions/test_multivariate.py → tests/distributions/test_multivariate.py b/.../tests/distributions/test_multivariate.py → tests/distributions/test_multivariate.py
diff --git a/pymc_experimental/tests/model/__init__.py → tests/model/__init__.py b/pymc_experimental/tests/model/__init__.py → tests/model/__init__.py
diff --git a/...mental/tests/model/test_marginal_model.py → tests/model/test_marginal_model.py b/...mental/tests/model/test_marginal_model.py → tests/model/test_marginal_model.py
@@ -22,7 +22,7 @@
     is_conditional_dependent,
     marginalize,
 )
-from pymc_experimental.tests.utils import equal_computations_up_to_root
+from tests.utils import equal_computations_up_to_root
 
 
 @pytest.fixture

diff --git a/...xperimental/tests/model/test_model_api.py → tests/model/test_model_api.py b/...xperimental/tests/model/test_model_api.py → tests/model/test_model_api.py
diff --git a/...ests/model/transforms/test_autoreparam.py → tests/model/transforms/test_autoreparam.py b/...ests/model/transforms/test_autoreparam.py → tests/model/transforms/test_autoreparam.py
diff --git a/...experimental/tests/statespace/__init__.py → tests/statespace/__init__.py b/...experimental/tests/statespace/__init__.py → tests/statespace/__init__.py
diff --git a/...rimental/tests/statespace/test_SARIMAX.py → tests/statespace/test_SARIMAX.py b/...rimental/tests/statespace/test_SARIMAX.py → tests/statespace/test_SARIMAX.py
@@ -17,10 +17,10 @@
     SARIMAX_STATE_STRUCTURES,
     SHORT_NAME_TO_LONG,
 )
-from pymc_experimental.tests.statespace.utilities.shared_fixtures import (  # pylint: disable=unused-import
+from tests.statespace.utilities.shared_fixtures import (  # pylint: disable=unused-import
     rng,
 )
-from pymc_experimental.tests.statespace.utilities.test_helpers import (
+from tests.statespace.utilities.test_helpers import (
     load_nile_test_data,
     make_stationary_params,
     simulate_from_numpy_model,

diff --git a/...erimental/tests/statespace/test_VARMAX.py → tests/statespace/test_VARMAX.py b/...erimental/tests/statespace/test_VARMAX.py → tests/statespace/test_VARMAX.py
@@ -11,7 +11,7 @@
 
 from pymc_experimental.statespace import BayesianVARMAX
 from pymc_experimental.statespace.utils.constants import SHORT_NAME_TO_LONG
-from pymc_experimental.tests.statespace.utilities.shared_fixtures import (  # pylint: disable=unused-import
+from tests.statespace.utilities.shared_fixtures import (  # pylint: disable=unused-import
     rng,
 )
 
@@ -25,7 +25,7 @@
 @pytest.fixture(scope="session")
 def data():
     df = pd.read_csv(
-        "pymc_experimental/tests/statespace/test_data/statsmodels_macrodata_processed.csv",
+        "tests/statespace/test_data/statsmodels_macrodata_processed.csv",
         index_col=0,
         parse_dates=True,
     ).astype(floatX)

diff --git a/...tests/statespace/test_coord_assignment.py → tests/statespace/test_coord_assignment.py b/...tests/statespace/test_coord_assignment.py → tests/statespace/test_coord_assignment.py
@@ -18,9 +18,7 @@
     NO_FREQ_INFO_WARNING,
     NO_TIME_INDEX_WARNING,
 )
-from pymc_experimental.tests.statespace.utilities.test_helpers import (
-    load_nile_test_data,
-)
+from tests.statespace.utilities.test_helpers import load_nile_test_data
 
 function_names = ["pandas_date_freq", "pandas_date_nofreq", "pandas_nodate", "numpy", "pytensor"]
 expected_warning = [

diff --git a/...al/tests/statespace/test_data/airpass.csv → tests/statespace/test_data/airpass.csv b/...al/tests/statespace/test_data/airpass.csv → tests/statespace/test_data/airpass.csv
diff --git a/...ts/statespace/test_data/airpassangers.csv → tests/statespace/test_data/airpassangers.csv b/...ts/statespace/test_data/airpassangers.csv → tests/statespace/test_data/airpassangers.csv
diff --git a/...ental/tests/statespace/test_data/nile.csv → tests/statespace/test_data/nile.csv b/...ental/tests/statespace/test_data/nile.csv → tests/statespace/test_data/nile.csv
diff --git a/..._data/statsmodels_macrodata_processed.csv → ..._data/statsmodels_macrodata_processed.csv b/..._data/statsmodels_macrodata_processed.csv → ..._data/statsmodels_macrodata_processed.csv
diff --git a/...al/tests/statespace/test_distributions.py → tests/statespace/test_distributions.py b/...al/tests/statespace/test_distributions.py → tests/statespace/test_distributions.py
@@ -17,10 +17,10 @@
     OBS_STATE_DIM,
     TIME_DIM,
 )
-from pymc_experimental.tests.statespace.utilities.shared_fixtures import (  # pylint: disable=unused-import
+from tests.statespace.utilities.shared_fixtures import (  # pylint: disable=unused-import
     rng,
 )
-from pymc_experimental.tests.statespace.utilities.test_helpers import (
+from tests.statespace.utilities.test_helpers import (
     delete_rvs_from_model,
     fast_eval,
     load_nile_test_data,

diff --git a/...al/tests/statespace/test_kalman_filter.py → tests/statespace/test_kalman_filter.py b/...al/tests/statespace/test_kalman_filter.py → tests/statespace/test_kalman_filter.py
@@ -13,10 +13,10 @@
     UnivariateFilter,
 )
 from pymc_experimental.statespace.filters.kalman_filter import BaseFilter
-from pymc_experimental.tests.statespace.utilities.shared_fixtures import (  # pylint: disable=unused-import
+from tests.statespace.utilities.shared_fixtures import (  # pylint: disable=unused-import
     rng,
 )
-from pymc_experimental.tests.statespace.utilities.test_helpers import (
+from tests.statespace.utilities.test_helpers import (
     get_expected_shape,
     get_sm_state_from_output_name,
     initialize_filter,

diff --git a/...l/tests/statespace/test_representation.py → tests/statespace/test_representation.py b/...l/tests/statespace/test_representation.py → tests/statespace/test_representation.py
@@ -6,11 +6,8 @@
 from numpy.testing import assert_allclose
 
 from pymc_experimental.statespace.core.representation import PytensorRepresentation
-from pymc_experimental.tests.statespace.utilities.shared_fixtures import TEST_SEED
-from pymc_experimental.tests.statespace.utilities.test_helpers import (
-    fast_eval,
-    make_test_inputs,
-)
+from tests.statespace.utilities.shared_fixtures import TEST_SEED
+from tests.statespace.utilities.test_helpers import fast_eval, make_test_inputs
 
 floatX = pytensor.config.floatX
 atol = 1e-12 if floatX == "float64" else 1e-6

diff --git a/...ental/tests/statespace/test_statespace.py → tests/statespace/test_statespace.py b/...ental/tests/statespace/test_statespace.py → tests/statespace/test_statespace.py
@@ -13,10 +13,10 @@
     MATRIX_NAMES,
     SMOOTHER_OUTPUT_NAMES,
 )
-from pymc_experimental.tests.statespace.utilities.shared_fixtures import (  # pylint: disable=unused-import
+from tests.statespace.utilities.shared_fixtures import (  # pylint: disable=unused-import
     rng,
 )
-from pymc_experimental.tests.statespace.utilities.test_helpers import (
+from tests.statespace.utilities.test_helpers import (
     fast_eval,
     load_nile_test_data,
     make_test_inputs,

diff --git a/...l/tests/statespace/test_statespace_JAX.py → tests/statespace/test_statespace_JAX.py b/...l/tests/statespace/test_statespace_JAX.py → tests/statespace/test_statespace_JAX.py
@@ -12,17 +12,14 @@
     MATRIX_NAMES,
     SMOOTHER_OUTPUT_NAMES,
 )
-from pymc_experimental.tests.statespace.test_statespace import (  # pylint: disable=unused-import
+from tests.statespace.test_statespace import (  # pylint: disable=unused-import
     exog_ss_mod,
-    make_statespace_mod,
     ss_mod,
 )
-from pymc_experimental.tests.statespace.utilities.shared_fixtures import (  # pylint: disable=unused-import
+from tests.statespace.utilities.shared_fixtures import (  # pylint: disable=unused-import
     rng,
 )
-from pymc_experimental.tests.statespace.utilities.test_helpers import (
-    load_nile_test_data,
-)
+from tests.statespace.utilities.test_helpers import load_nile_test_data
 
 pytest.importorskip("jax")
 pytest.importorskip("numpyro")

diff --git a/...ental/tests/statespace/test_structural.py → tests/statespace/test_structural.py b/...ental/tests/statespace/test_structural.py → tests/statespace/test_structural.py
@@ -24,10 +24,10 @@
     SHOCK_DIM,
     SHORT_NAME_TO_LONG,
 )
-from pymc_experimental.tests.statespace.utilities.shared_fixtures import (  # pylint: disable=unused-import
+from tests.statespace.utilities.shared_fixtures import (  # pylint: disable=unused-import
     rng,
 )
-from pymc_experimental.tests.statespace.utilities.test_helpers import (
+from tests.statespace.utilities.test_helpers import (
     assert_pattern_repeats,
     simulate_from_numpy_model,
     unpack_symbolic_matrices_with_params,

diff --git a/...al/tests/statespace/utilities/__init__.py → tests/statespace/utilities/__init__.py b/...al/tests/statespace/utilities/__init__.py → tests/statespace/utilities/__init__.py
diff --git a/...s/statespace/utilities/shared_fixtures.py → ...s/statespace/utilities/shared_fixtures.py b/...s/statespace/utilities/shared_fixtures.py → ...s/statespace/utilities/shared_fixtures.py
diff --git a/...space/utilities/statsmodel_local_level.py → ...space/utilities/statsmodel_local_level.py b/...space/utilities/statsmodel_local_level.py → ...space/utilities/statsmodel_local_level.py
diff --git a/...ests/statespace/utilities/test_helpers.py → tests/statespace/utilities/test_helpers.py b/...ests/statespace/utilities/test_helpers.py → tests/statespace/utilities/test_helpers.py
@@ -11,17 +11,15 @@
     MATRIX_NAMES,
     SHORT_NAME_TO_LONG,
 )
-from pymc_experimental.tests.statespace.utilities.statsmodel_local_level import (
-    LocalLinearTrend,
-)
+from tests.statespace.utilities.statsmodel_local_level import LocalLinearTrend
 
 floatX = pytensor.config.floatX
 
 
 def load_nile_test_data():
     from importlib.metadata import version
 
-    nile = pd.read_csv("pymc_experimental/tests/statespace/test_data/nile.csv", dtype={"x": floatX})
+    nile = pd.read_csv("tests/statespace/test_data/nile.csv", dtype={"x": floatX})
     major, minor, rev = map(int, version("pandas").split("."))
     if major >= 2 and minor >= 2 and rev >= 0:
         freq_str = "YS-JAN"

diff --git a/pymc_experimental/tests/test_blackjax_smc.py → tests/test_blackjax_smc.py b/pymc_experimental/tests/test_blackjax_smc.py → tests/test_blackjax_smc.py
diff --git a/...tal/tests/test_histogram_approximation.py → tests/test_histogram_approximation.py b/...tal/tests/test_histogram_approximation.py → tests/test_histogram_approximation.py
diff --git a/pymc_experimental/tests/test_laplace.py → tests/test_laplace.py b/pymc_experimental/tests/test_laplace.py → tests/test_laplace.py
diff --git a/pymc_experimental/tests/test_linearmodel.py → tests/test_linearmodel.py b/pymc_experimental/tests/test_linearmodel.py → tests/test_linearmodel.py
diff --git a/..._experimental/tests/test_model_builder.py → tests/test_model_builder.py b/..._experimental/tests/test_model_builder.py → tests/test_model_builder.py
diff --git a/pymc_experimental/tests/test_pathfinder.py → tests/test_pathfinder.py b/pymc_experimental/tests/test_pathfinder.py → tests/test_pathfinder.py
diff --git a/...perimental/tests/test_pivoted_cholesky.py → tests/test_pivoted_cholesky.py b/...perimental/tests/test_pivoted_cholesky.py → tests/test_pivoted_cholesky.py
diff --git a/...perimental/tests/test_prior_from_trace.py → tests/test_prior_from_trace.py b/...perimental/tests/test_prior_from_trace.py → tests/test_prior_from_trace.py
diff --git a/pymc_experimental/tests/test_splines.py → tests/test_splines.py b/pymc_experimental/tests/test_splines.py → tests/test_splines.py
diff --git a/pymc_experimental/tests/utils.py → tests/utils.py b/pymc_experimental/tests/utils.py → tests/utils.py