pymc.testing.mock_sample

pymc.testing.mock_sample(draws=10, **kwargs)

Mock pymc.sample() with pymc.sample_prior_predictive().

Useful for testing models that use pm.sample without running MCMC sampling.

Examples

Using mock_sample with pytest

Note

Use pymc.testing.mock_sample_setup_and_teardown() directly for pytest fixtures.

import pytest

import pymc as pm
from pymc.testing import mock_sample

@pytest.fixture(scope="module")
def mock_pymc_sample():
    original_sample = pm.sample
    pm.sample = mock_sample

    yield

    pm.sample = original_sample