pyEMU is a set of python modules for interfacing with PEST and PEST++
python modules for model-independent FOSM (first-order, second-moment) (a.k.a linear-based, a.k.a. Bayes linear)
uncertainty analyses and data-worth analyses, non-linear uncertainty analyses and interfacing with PEST and PEST++.
pyEMU also has a pure python (pandas and numpy) implementation of ordinary kriging for geostatistical interpolation and support for generating high-dimensional PEST(++) model interfaces, including support for (very) high-dimensional ensemble generation and handling
Complete user's guide:
The pyEMU documentation is being treated as a first-class citizen! Also see the example notebooks in the repo.
What is pyEMU?
pyEMU is a set of python modules for model-independent, user-friendly, computer model uncertainty analysis. pyEMU is tightly coupled to the open-source suite PEST (Doherty 2010a and 2010b, and Doherty and other, 2010) and PEST++ (Welter and others, 2015, Welter and other, 2012), which are tools for model-independent parameter estimation. However, pyEMU can be used with generic array objects, such as numpy ndarrays.
Several equations are implemented, including Schur's complement for conditional uncertainty propagation (a.k.a. Bayes Linear estimation) (the foundation of the PREDUNC suite from PEST) and error variance analysis (the foundation of the PREDVAR suite of PEST). pyEMU has easy-to-use routines for parameter and data worth analyses, which estimate how increased parameter knowledge and/or additional data effect forecast uncertainty in linear, Bayesian framework. Support is also provided for high-dimensional Monte Carlo analyses via
ParameterEnsemble class, including the null-space monte carlo approach of Tonkin and Doherty (2009); these ensemble classes also play nicely with PESTPP-IES.
pyEMU also includes lots of functionality for dealing with PEST(++) datasets, such as:
- manipulation of PEST control files, including the use of pandas for sophisticated editing of the parameter data and observation data sections
- creation of PEST control files from instruction and template files
- going between site sample files and pandas dataframes - really cool for observation processing
- easy-to-use observation (re)weighting via residuals or user-defined functions
- handling Jacobian and covariance matrices, including functionality to go between binary and ASCII matrices, reading and writing PEST uncertainty files. Covariance matrices can be instantiated from relevant control file sections, such as parameter bounds or observation weights. The base Matrix class overloads most common linear algebra operators so that operations are automatically aligned by row and column name. Builtin SVD is also included in all Matrix instances.
- geostatistics including geostatistical structure support, reading and writing PEST structure files and creating covariance matrices implied by nested geostatistical structures, and ordinary kriging (in the utils.geostats.OrdinaryKrige object), which replicates the functionality of pest utility
- composite scaled sensitivity calculations
- calculation of correlation coefficient matrix from a given covariance matrix
- Karhunen-Loeve-based parameterization as an alternative to pilot points for spatially-distributed parameter fields
- a helper functions to start a group of tcp/ip workers on a local machine for parallel PEST++/BeoPEST runs
- full support for prior information equations in control files
- preferred differencing prior information equations where the weights are based on the Pearson correlation coefficient
- verification-based tests based on results from several PEST utilities
Version => 1.1 includes the
PstFrom setup class to support generating PEST(++) interfaces in the 100,000 to 1,000,000 parameter range with all the bells and whistles. A publication documenting the
PstFrom class can be found here:
A publication documenting pyEMU and an example application can be found here:
pyEMU was originally developed with support from the U.S. Geological Survey. The New Zealand Strategic Science Investment Fund as part of GNS Science’s (https://www.gns.cri.nz/) Groundwater Research Programme has also funded contributions 2018-present. Intera, Inc. has also provided funding for pyEMU development and support
Several example ipython notebooks are provided to demonstrate typical workflows for FOSM parameter and forecast uncertainty analysis as well as techniques to investigate parameter contributions to forecast uncertainty and observation data worth. Example models include the Henry saltwater intrusion problem (Henry 1964) and the model of Freyberg (1988)
There is a whole world of detailed learning material for script-based approaches to parameter estimation and uncertainty quantification using PEST(++) at https://github.com/gmdsi/GMDSI_notebooks. These are and excellent resource for people picking up Pyemu for the first time and for those needing to revisit elements.
How to get started with pyEMU
pyEMU is available through pyPI and conda. To install pyEMU type:
>>>conda install -c conda-forge pyemu
>>>pip install pyemu
pandas. For plotting,
flopy to take advantage of the auto interface construction
After pyEMU is installed, the PEST++ software suite can be installed for your operating system using the command:
See documentation for more information.
Found a bug? Got a smart idea? Contributions welcome.
Feel free to raise and issue or submit a pull request.
pyEMU CI testing, using GitHub actions, has recently been switched over to run with pytest. We make use of pytest-xdist for parallel execution. Some notes that might be helpful for building your PR and testing:
- Test files are in ./autotest
- Pytest settings are in ./autotest/conftest.py and ./autotest/pytest.ini
- Currently, files ending
- Functions starting
- ipython notebooks in .examples are also run
- As tests are run in parallel, where tests require read/write access to files it is safest to sandbox runs.
Pytest has a built-in fixture
tmp_paththat can help with this. Setting optional argument
--basetempcan be helpful for accessing the locally run files.
Running test locally
To be able to make clean use of pytests fixture decorators etc.,
it is recommended to run local tests through
pytest (rather than use from script execution and commenting in
main block). For e.g.:
Run all tests:
pytest --basetemp=runner autotest
with pytest-xdist, local runs can be parallelized:
pytest --basetemp=runner -n auto autotest
Run all tests in a file:
pytest --basetemp=runner -n auto autotest/testfile_tests.py
Run a specific test [
pytest --basetemp=runner autotest/testfile_tests.py::this_test
Using an IDE:
Most modern, feature-rich editors and IDEs support launching pytest within debug or run consoles.
Some might need "encouraging" to recognise the non-standard test tags used in this library.
For example, in pycharm, to support click-and-run testing, the
pytest-imp plugin is required to
pickup test functions that end with
_test (a nosetest hangover in pyEMU).
Release history Release notifications | RSS feed
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.