Skip to content

Commit 64f0f1b

Browse files
committed
Move unit test
1 parent 4429e46 commit 64f0f1b

File tree

2 files changed

+26
-27
lines changed

2 files changed

+26
-27
lines changed

test/test_compilation.py

Lines changed: 26 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,12 @@
99

1010
import pytest
1111

12-
from cmdstanpy.compilation import CompilerOptions, format_stan_file
12+
from cmdstanpy.compilation import (
13+
CompilerOptions,
14+
format_stan_file,
15+
resolve_cpp_options,
16+
resolve_stanc_options,
17+
)
1318

1419
HERE = os.path.dirname(os.path.abspath(__file__))
1520
DATAFILES_PATH = os.path.join(HERE, 'data')
@@ -225,3 +230,23 @@ def test_model_format_options() -> None:
225230
formatted = sys_stdout.getvalue()
226231
assert formatted.count('{') == 3
227232
assert formatted.count('(') == 1
233+
234+
235+
def test_compilation_options_resolution() -> None:
236+
out = resolve_cpp_options(None, multithreading=False)
237+
assert not out
238+
out = resolve_cpp_options(None, multithreading=True)
239+
assert out == {"STAN_THREADS": "TRUE"}
240+
out = resolve_cpp_options({"STAN_THREADS": ""}, multithreading=True)
241+
assert out == {"STAN_THREADS": ""}
242+
out = resolve_cpp_options({"STAN_OPENCL": "TRUE"}, multithreading=True)
243+
assert out == {"STAN_THREADS": "TRUE", "STAN_OPENCL": "TRUE"}
244+
245+
out = resolve_stanc_options(None, stanc_optimizations=False)
246+
assert not out
247+
out = resolve_stanc_options(None, stanc_optimizations=True)
248+
assert out == {"O": 1}
249+
out = resolve_stanc_options({"O": 0}, stanc_optimizations=True)
250+
assert out == {"O": 0}
251+
out = resolve_stanc_options({"O": "experimental"}, stanc_optimizations=True)
252+
assert out == {"O": "experimental"}

test/test_model.py

Lines changed: 0 additions & 26 deletions
Original file line numberDiff line numberDiff line change
@@ -471,29 +471,3 @@ def test_diagnose() -> None:
471471
require_gradients_ok=False,
472472
)
473473
assert np.abs(gradients["error"]).max() > 1e-3
474-
475-
476-
def test_compilation_options_resolution() -> None:
477-
model = CmdStanModel(stan_file=BERN_STAN)
478-
479-
out = model._resolve_cpp_options(None, multithreading=False)
480-
assert not out
481-
out = model._resolve_cpp_options(None, multithreading=True)
482-
assert out == {"STAN_THREADS": "TRUE"}
483-
out = model._resolve_cpp_options({"STAN_THREADS": ""}, multithreading=True)
484-
assert out == {"STAN_THREADS": ""}
485-
out = model._resolve_cpp_options(
486-
{"STAN_OPENCL": "TRUE"}, multithreading=True
487-
)
488-
assert out == {"STAN_THREADS": "TRUE", "STAN_OPENCL": "TRUE"}
489-
490-
out = model._resolve_stanc_options(None, stanc_optimizations=False)
491-
assert not out
492-
out = model._resolve_stanc_options(None, stanc_optimizations=True)
493-
assert out == {"O": 1}
494-
out = model._resolve_stanc_options({"O": 0}, stanc_optimizations=True)
495-
assert out == {"O": 0}
496-
out = model._resolve_stanc_options(
497-
{"O": "experimental"}, stanc_optimizations=True
498-
)
499-
assert out == {"O": "experimental"}

0 commit comments

Comments
 (0)