Skip to content

Commit c8859ed

Browse files
jessegrabowskijessegrabowski
authored
Reorganize statespace tests (#480)
* Rearrange statespace test folder * Update ci
1 parent bb858a4 commit c8859ed

25 files changed

+97
-92
lines changed

‎.github/workflows/test.yml‎

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -28,9 +28,9 @@ jobs:
2828
python-version: ["3.10"]
2929
test-subset:
3030
- tests/model
31-
- tests/statespace/test_statespace.py
32-
- tests/statespace/test_kalman_filter.py
33-
- tests/statespace --ignore tests/statespace/test_statespace.py --ignore tests/statespace/test_kalman_filter.py
31+
- tests/statespace/core/test_statespace.py
32+
- tests/statespace/filters/test_kalman_filter.py
33+
- tests/statespace --ignore tests/statespace/core/test_statespace.py --ignore tests/statespace/filters/test_kalman_filter.py
3434
- tests/distributions
3535
- tests --ignore tests/model --ignore tests/statespace --ignore tests/distributions
3636
fail-fast: false
@@ -71,9 +71,9 @@ jobs:
7171
python-version: ["3.12"]
7272
test-subset:
7373
- tests/model
74-
- tests/statespace/test_statespace.py
75-
- tests/statespace/test_kalman_filter.py
76-
- tests/statespace --ignore tests/statespace/test_statespace.py --ignore tests/statespace/test_kalman_filter.py
74+
- tests/statespace/core/test_statespace.py
75+
- tests/statespace/filters/test_kalman_filter.py
76+
- tests/statespace --ignore tests/statespace/core/test_statespace.py --ignore tests/statespace/filters/test_kalman_filter.py
7777
- tests/distributions
7878
- tests --ignore tests/model --ignore tests/statespace --ignore tests/distributions
7979
fail-fast: false

‎notebooks/Exponential Trend Smoothing.ipynb‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3559,7 +3559,7 @@
35593559
],
35603560
"source": [
35613561
"df = (\n",
3562-
" pd.read_csv(\"../tests/statespace/test_data/airpassangers.csv\", parse_dates=[\"Month\"])\n",
3562+
" pd.read_csv(\"../tests/statespace/_data/airpassangers.csv\", parse_dates=[\"Month\"])\n",
35633563
" .rename(columns={\"Month\": \"time\", \"#Passengers\": \"passangers\"})\n",
35643564
" .set_index(\"time\")\n",
35653565
" .astype(float)\n",

‎notebooks/SARMA Example.ipynb‎

Lines changed: 29 additions & 29 deletions
Original file line numberDiff line numberDiff line change
@@ -337,17 +337,17 @@
337337
"</pre>\n"
338338
],
339339
"text/plain": [
340-
"\u001b[3m Model Requirements \u001b[0m\n",
340+
"\u001B[3m Model Requirements \u001B[0m\n",
341341
" \n",
342-
" \u001b[1m \u001b[0m\u001b[1mVariable \u001b[0m\u001b[1m \u001b[0m \u001b[1m \u001b[0m\u001b[1mShape\u001b[0m\u001b[1m \u001b[0m \u001b[1m \u001b[0m\u001b[1mConstraints\u001b[0m\u001b[1m \u001b[0m \u001b[1m \u001b[0m\u001b[1m Dimensions\u001b[0m\u001b[1m \u001b[0m \n",
342+
" \u001B[1m \u001B[0m\u001B[1mVariable \u001B[0m\u001B[1m \u001B[0m \u001B[1m \u001B[0m\u001B[1mShape\u001B[0m\u001B[1m \u001B[0m \u001B[1m \u001B[0m\u001B[1mConstraints\u001B[0m\u001B[1m \u001B[0m \u001B[1m \u001B[0m\u001B[1m Dimensions\u001B[0m\u001B[1m \u001B[0m \n",
343343
" ───────────────────────────────────────────────── \n",
344-
" ar_params \u001b[1m(\u001b[0m\u001b[1;36m1\u001b[0m,\u001b[1m)\u001b[0m \u001b[3;35mNone\u001b[0m \u001b[1m(\u001b[0m\u001b[32m'ar_lag'\u001b[0m,\u001b[1m)\u001b[0m \n",
345-
" ma_params \u001b[1m(\u001b[0m\u001b[1;36m1\u001b[0m,\u001b[1m)\u001b[0m \u001b[3;35mNone\u001b[0m \u001b[1m(\u001b[0m\u001b[32m'ma_lag'\u001b[0m,\u001b[1m)\u001b[0m \n",
346-
" sigma_state \u001b[3;35mNone\u001b[0m Positive \u001b[3;35mNone\u001b[0m \n",
344+
" ar_params \u001B[1m(\u001B[0m\u001B[1;36m1\u001B[0m,\u001B[1m)\u001B[0m \u001B[3;35mNone\u001B[0m \u001B[1m(\u001B[0m\u001B[32m'ar_lag'\u001B[0m,\u001B[1m)\u001B[0m \n",
345+
" ma_params \u001B[1m(\u001B[0m\u001B[1;36m1\u001B[0m,\u001B[1m)\u001B[0m \u001B[3;35mNone\u001B[0m \u001B[1m(\u001B[0m\u001B[32m'ma_lag'\u001B[0m,\u001B[1m)\u001B[0m \n",
346+
" sigma_state \u001B[3;35mNone\u001B[0m Positive \u001B[3;35mNone\u001B[0m \n",
347347
" \n",
348-
"\u001b[2;3mThese parameters should be assigned priors inside a\u001b[0m\n",
349-
"\u001b[2;3m PyMC model block before calling the \u001b[0m\n",
350-
"\u001b[2;3m build_statespace_graph method. \u001b[0m\n"
348+
"\u001B[2;3mThese parameters should be assigned priors inside a\u001B[0m\n",
349+
"\u001B[2;3m PyMC model block before calling the \u001B[0m\n",
350+
"\u001B[2;3m build_statespace_graph method. \u001B[0m\n"
351351
]
352352
},
353353
"metadata": {},
@@ -1805,18 +1805,18 @@
18051805
"</pre>\n"
18061806
],
18071807
"text/plain": [
1808-
"\u001b[3m Model Requirements \u001b[0m\n",
1808+
"\u001B[3m Model Requirements \u001B[0m\n",
18091809
" \n",
1810-
" \u001b[1m \u001b[0m\u001b[1mVariable \u001b[0m\u001b[1m \u001b[0m \u001b[1m \u001b[0m\u001b[1mShape\u001b[0m\u001b[1m \u001b[0m \u001b[1m \u001b[0m\u001b[1mConstraints\u001b[0m\u001b[1m \u001b[0m \u001b[1m \u001b[0m\u001b[1m Dimensions\u001b[0m\u001b[1m \u001b[0m \n",
1810+
" \u001B[1m \u001B[0m\u001B[1mVariable \u001B[0m\u001B[1m \u001B[0m \u001B[1m \u001B[0m\u001B[1mShape\u001B[0m\u001B[1m \u001B[0m \u001B[1m \u001B[0m\u001B[1mConstraints\u001B[0m\u001B[1m \u001B[0m \u001B[1m \u001B[0m\u001B[1m Dimensions\u001B[0m\u001B[1m \u001B[0m \n",
18111811
" ───────────────────────────────────────────────── \n",
1812-
" ar_params \u001b[1m(\u001b[0m\u001b[1;36m1\u001b[0m,\u001b[1m)\u001b[0m \u001b[3;35mNone\u001b[0m \u001b[1m(\u001b[0m\u001b[32m'ar_lag'\u001b[0m,\u001b[1m)\u001b[0m \n",
1813-
" ma_params \u001b[1m(\u001b[0m\u001b[1;36m1\u001b[0m,\u001b[1m)\u001b[0m \u001b[3;35mNone\u001b[0m \u001b[1m(\u001b[0m\u001b[32m'ma_lag'\u001b[0m,\u001b[1m)\u001b[0m \n",
1814-
" sigma_state \u001b[3;35mNone\u001b[0m Positive \u001b[3;35mNone\u001b[0m \n",
1815-
" sigma_obs \u001b[3;35mNone\u001b[0m Positive \u001b[3;35mNone\u001b[0m \n",
1812+
" ar_params \u001B[1m(\u001B[0m\u001B[1;36m1\u001B[0m,\u001B[1m)\u001B[0m \u001B[3;35mNone\u001B[0m \u001B[1m(\u001B[0m\u001B[32m'ar_lag'\u001B[0m,\u001B[1m)\u001B[0m \n",
1813+
" ma_params \u001B[1m(\u001B[0m\u001B[1;36m1\u001B[0m,\u001B[1m)\u001B[0m \u001B[3;35mNone\u001B[0m \u001B[1m(\u001B[0m\u001B[32m'ma_lag'\u001B[0m,\u001B[1m)\u001B[0m \n",
1814+
" sigma_state \u001B[3;35mNone\u001B[0m Positive \u001B[3;35mNone\u001B[0m \n",
1815+
" sigma_obs \u001B[3;35mNone\u001B[0m Positive \u001B[3;35mNone\u001B[0m \n",
18161816
" \n",
1817-
"\u001b[2;3mThese parameters should be assigned priors inside a\u001b[0m\n",
1818-
"\u001b[2;3m PyMC model block before calling the \u001b[0m\n",
1819-
"\u001b[2;3m build_statespace_graph method. \u001b[0m\n"
1817+
"\u001B[2;3mThese parameters should be assigned priors inside a\u001B[0m\n",
1818+
"\u001B[2;3m PyMC model block before calling the \u001B[0m\n",
1819+
"\u001B[2;3m build_statespace_graph method. \u001B[0m\n"
18201820
]
18211821
},
18221822
"metadata": {},
@@ -2331,7 +2331,7 @@
23312331
],
23322332
"source": [
23332333
"airpass = pd.read_csv(\n",
2334-
" \"../tests/statespace/test_data/airpass.csv\",\n",
2334+
" \"../tests/statespace/_data/airpass.csv\",\n",
23352335
" parse_dates=True,\n",
23362336
" date_format=\"%Y %b\",\n",
23372337
" index_col=0,\n",
@@ -2414,20 +2414,20 @@
24142414
"</pre>\n"
24152415
],
24162416
"text/plain": [
2417-
"\u001b[3m Model Requirements \u001b[0m\n",
2417+
"\u001B[3m Model Requirements \u001B[0m\n",
24182418
" \n",
2419-
" \u001b[1m \u001b[0m\u001b[1mVariable \u001b[0m\u001b[1m \u001b[0m \u001b[1m \u001b[0m\u001b[1mShape \u001b[0m\u001b[1m \u001b[0m \u001b[1m \u001b[0m\u001b[1mConstraints \u001b[0m\u001b[1m \u001b[0m \u001b[1m \u001b[0m\u001b[1m Dimensions\u001b[0m\u001b[1m \u001b[0m \n",
2419+
" \u001B[1m \u001B[0m\u001B[1mVariable \u001B[0m\u001B[1m \u001B[0m \u001B[1m \u001B[0m\u001B[1mShape \u001B[0m\u001B[1m \u001B[0m \u001B[1m \u001B[0m\u001B[1mConstraints \u001B[0m\u001B[1m \u001B[0m \u001B[1m \u001B[0m\u001B[1m Dimensions\u001B[0m\u001B[1m \u001B[0m \n",
24202420
" ───────────────────────────────────────────────────────────────────────────────── \n",
2421-
" x0 \u001b[1m(\u001b[0m\u001b[1;36m28\u001b[0m,\u001b[1m)\u001b[0m \u001b[1m(\u001b[0m\u001b[32m'state'\u001b[0m,\u001b[1m)\u001b[0m \n",
2422-
" P0 \u001b[1m(\u001b[0m\u001b[1;36m28\u001b[0m, \u001b[1;36m28\u001b[0m\u001b[1m)\u001b[0m Positive Semi-definite \u001b[1m(\u001b[0m\u001b[32m'state'\u001b[0m, \u001b[32m'state_aux'\u001b[0m\u001b[1m)\u001b[0m \n",
2423-
" ar_params \u001b[1m(\u001b[0m\u001b[1;36m2\u001b[0m,\u001b[1m)\u001b[0m \u001b[3;35mNone\u001b[0m \u001b[1m(\u001b[0m\u001b[32m'ar_lag'\u001b[0m,\u001b[1m)\u001b[0m \n",
2424-
" ma_params \u001b[1m(\u001b[0m\u001b[1;36m2\u001b[0m,\u001b[1m)\u001b[0m \u001b[3;35mNone\u001b[0m \u001b[1m(\u001b[0m\u001b[32m'ma_lag'\u001b[0m,\u001b[1m)\u001b[0m \n",
2425-
" seasonal_ar_params \u001b[1m(\u001b[0m\u001b[1;36m2\u001b[0m,\u001b[1m)\u001b[0m \u001b[3;35mNone\u001b[0m \u001b[1m(\u001b[0m\u001b[32m'seasonal_ar_lag'\u001b[0m,\u001b[1m)\u001b[0m \n",
2426-
" seasonal_ma_params \u001b[1m(\u001b[0m\u001b[1;36m2\u001b[0m,\u001b[1m)\u001b[0m \u001b[3;35mNone\u001b[0m \u001b[1m(\u001b[0m\u001b[32m'seasonal_ma_lag'\u001b[0m,\u001b[1m)\u001b[0m \n",
2427-
" sigma_state \u001b[3;35mNone\u001b[0m Positive \u001b[3;35mNone\u001b[0m \n",
2421+
" x0 \u001B[1m(\u001B[0m\u001B[1;36m28\u001B[0m,\u001B[1m)\u001B[0m \u001B[1m(\u001B[0m\u001B[32m'state'\u001B[0m,\u001B[1m)\u001B[0m \n",
2422+
" P0 \u001B[1m(\u001B[0m\u001B[1;36m28\u001B[0m, \u001B[1;36m28\u001B[0m\u001B[1m)\u001B[0m Positive Semi-definite \u001B[1m(\u001B[0m\u001B[32m'state'\u001B[0m, \u001B[32m'state_aux'\u001B[0m\u001B[1m)\u001B[0m \n",
2423+
" ar_params \u001B[1m(\u001B[0m\u001B[1;36m2\u001B[0m,\u001B[1m)\u001B[0m \u001B[3;35mNone\u001B[0m \u001B[1m(\u001B[0m\u001B[32m'ar_lag'\u001B[0m,\u001B[1m)\u001B[0m \n",
2424+
" ma_params \u001B[1m(\u001B[0m\u001B[1;36m2\u001B[0m,\u001B[1m)\u001B[0m \u001B[3;35mNone\u001B[0m \u001B[1m(\u001B[0m\u001B[32m'ma_lag'\u001B[0m,\u001B[1m)\u001B[0m \n",
2425+
" seasonal_ar_params \u001B[1m(\u001B[0m\u001B[1;36m2\u001B[0m,\u001B[1m)\u001B[0m \u001B[3;35mNone\u001B[0m \u001B[1m(\u001B[0m\u001B[32m'seasonal_ar_lag'\u001B[0m,\u001B[1m)\u001B[0m \n",
2426+
" seasonal_ma_params \u001B[1m(\u001B[0m\u001B[1;36m2\u001B[0m,\u001B[1m)\u001B[0m \u001B[3;35mNone\u001B[0m \u001B[1m(\u001B[0m\u001B[32m'seasonal_ma_lag'\u001B[0m,\u001B[1m)\u001B[0m \n",
2427+
" sigma_state \u001B[3;35mNone\u001B[0m Positive \u001B[3;35mNone\u001B[0m \n",
24282428
" \n",
2429-
"\u001b[2;3mThese parameters should be assigned priors inside a PyMC model block before calling\u001b[0m\n",
2430-
"\u001b[2;3m the build_statespace_graph method. \u001b[0m\n"
2429+
"\u001B[2;3mThese parameters should be assigned priors inside a PyMC model block before calling\u001B[0m\n",
2430+
"\u001B[2;3m the build_statespace_graph method. \u001B[0m\n"
24312431
]
24322432
},
24332433
"metadata": {},

0 commit comments

Comments
 (0)