Move tests/ to project root (#361)

Author: jessegrabowski

.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:

codecov.yml

@@ -22,7 +22,7 @@ coverage:
         base: auto
 
 ignore:
-  - "pymc_experimental/tests/*"
+  - "tests/*"
 
 comment:
   layout: "reach, diff, flags, files"

pymc_experimental/tests/__init__.py renamed to tests/__init__.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

pymc_experimental/tests/distributions/__init__.py renamed to tests/distributions/__init__.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,

pymc_experimental/tests/distributions/test_continuous.py renamed to tests/distributions/test_continuous.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)

pymc_experimental/tests/distributions/test_discrete.py renamed to tests/distributions/test_discrete.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 = [

pymc_experimental/tests/distributions/test_discrete_markov_chain.py renamed to tests/distributions/test_discrete_markov_chain.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,

pymc_experimental/tests/distributions/test_multivariate.py renamed to tests/distributions/test_multivariate.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,

pymc_experimental/tests/model/__init__.py renamed to tests/model/__init__.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

pymc_experimental/tests/model/test_marginal_model.py renamed to tests/model/test_marginal_model.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,

pymc_experimental/tests/model/test_model_api.py renamed to tests/model/test_model_api.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")

pymc_experimental/tests/model/transforms/test_autoreparam.py renamed to tests/model/transforms/test_autoreparam.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,

pymc_experimental/tests/statespace/__init__.py renamed to tests/statespace/__init__.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"