User manual¶
pytest-wdl is a plugin for the pytest unit testing framework that enables testing of workflows written in Workflow Description Language. Test workflow inputs and expected outputs are configured in a test_data.json
file. Workflows are run by one or more executors. By default, actual and expected outputs are compared by MD5 hash, but data type-specific comparisons are provided. Data types and executors are pluggable and can be provided via third-party packages.
Fixtures¶
All functionality of pytest-wdl is provided via fixtures. As long as pytest-wdl is in your PYTHONPATH
, its fixtures will be discovered and made available when you run pytest.
The two most important fixtures are:
workflow_data: Provides access to data files for use as inputs to a workflow, and for comparing to workflow output.
workflow_runner: Given a WDL workflow, inputs, and expected outputs, runs the workflow using one or more executors and compares actual and expected outputs.
There are also several additional fixtures used for configuration of the two main fixtures. In most cases, the default values returned by these fixtures “just work.” However, if you need to override the defaults, you may do so either directly within your test modules, or in a conftest.py file.
Test data¶
Typically, workflows require inputs and generate outputs. Beyond simply ensuring that a workflow runs successfully, we often want to additionally test that it reproducibly generates the same results given the same inputs.
Test inputs and outputs are configured in a test_data.json
file that is stored in the same directory as the test module. This file has one entry for each input/output. Primitive types map as expected from JSON to Python to WDL. For example, the following test_data.json
file defines an integer input that is loaded as a Python int
and then maps to the WDL Integer
type when passed as an input parameter to a workflow:
{
"input_int": 42
}
Files¶
For file inputs and outputs, pytest-wdl offers several different options. Test data files may be located remotely (identified by a URL), located within the test directory (using the folder hierarchy established by the datadir-ng plugin), located at an arbitrary local path, or defined by specifying the file contents directly within the JSON file. Files that do not already exist locally are localized on-demand and stored in the cache directory.
Some additional options are available only for expected outputs, in order to specify how they should be compared to the actual outputs.
Below is an example test_data.json
file that demonstrates different ways to define input and output data files:
{
"bam": {
"url": "http://example.com/my.bam",
"http_headers": {
"auth_token": "TOKEN"
}
},
"reference": {
"path": "${REFERENCE_DIR}/chr22.fa"
},
"sample": {
"path": "samples.vcf",
"contents": "sample1\nsample2"
},
"output_vcf": {
"name": "output.vcf",
"type": "vcf",
"allowed_diff_lines": 2
}
}
The available keys for configuring file inputs/outputs are:
name
: Filename to use when localizing the file; when none ofurl
,path
, orcontents
are defined,name
is also used to search for the data file within the tests directory, using the same directory structure defined by the datadir-ng fixture.path
: The local path to the file. If the path does not already exist, the file will be localized to this path. Typically, this is defined as a relative path that will be prefixed with the cache directory path. Environment variables can be used to enable the user to configure an environment-specific path.env
: The name of an environment variable in which to look up the local path of the file.url
: A URL that can be resolved by urllib.http_headers
: Optional dict mapping header names to values. These headers are used for file download requests. Keys are header names and values are either strings (environment variable name) or mappings with the following keys:env
: The name of an environment variable in which to look up the header value.value
: The header value; only used if an environment variable is not specified or is unset.
contents
: The contents of the file, specified as a string. The file is written topath
the first time it is requested.
In addition, the following keys are recognized for output files only:
type
: The file type. This is optional and only needs to be provided for certain types of files that are handled specially for the sake of comparison.allowed_diff_lines
: Optional and only used for outputs comparison. If ‘0’ or not specified, it is assumed that the expected and actual outputs are identical.
Data Types¶
When comparing actual and expected outputs, the “type” of the expected output is used to determine how the files are compared. If no type is specified, then the type is assumed to be “default.”
Available types:
default: The default type if one is not specified.
It can handle raw text files, as well as gzip compressed files.
If
allowed_diff_lines
is 0 or not specified, then the files are compared by their MD5 hashes.If
allowed_diff_lines
is > 0, the files are converted to text and compared using the linuxdiff
tool.
vcf: During comparison, headers are ignored, as are the QUAL, INFO, and FORMAT columns; for sample columns, only the first sample column is compared between files, and only the genotype values for that sample.
bam*:
BAM is converted to SAM and headers are ignored.
Replaces random UNSET-\w*\b type IDs that samtools often adds.
* requires extra dependencies to be installed, see Installing Data Type Plugins
Executors¶
An Executor is a wrapper around a WDL workflow execution engine that prepares inputs, runs the tool, captures outputs, and handles errors. Currently, Cromwell is the only supported executor, but aternative executors can be implemented as plugins.
The workflow_runner
fixture is a callable that runs the workflow using the executor. It takes one required arguments and several additional optional arguments:
wdl_script
: Required; the WDL script to execute. The path must either be absolute or (more commonly) relative to the project root.workflow_name
: The name of the workflow to execute in the WDL script. If not specified, it is assumed that the workflow has the same name as the WDL file (without the “.wdl” extension).inputs
: Dict that will be serialized to JSON and provided to Cromwell as the workflow inputs. If not specified, the workflow must not have any required inputs.expected
: Dict mapping output parameter names to expected values. Any workflow outputs that are not specified are ignored. This is an optional parameter and can be omitted if, for example, you only want to test that the workflow completes successfully.
Executor-specific options¶
Cromwell¶
inputs_file
: Specify the inputs.json file to use, or the path to the inputs.json file to write, instead of a temp file.imports_file
: Specify the imports file to use, or the path to the imports zip file to write, instead of a temp file.java_args
: Override the default Java arguments.cromwell_args
: Override the default Cromwell arguments.
Configuration¶
pytest-wdl has two levels of configuration:
Project-specific configuration, which generally deals with the structure of the project, and may require customization if the structure of your project differs substantially from what is expected, but also encompases executor-specific configuration.
Environment-specific configuration, which generally deals with idiosyncrasies of the local environment.
Project-specific configuration¶
Configuration at the project level is handled by overriding fixtures, either in the test module or in a top-level conftest.py file. The following fixtures may be overridden:
fixture name | scope | description | default |
---|---|---|---|
project_root_files |
module | List of filenames that are found in the project root directory. | ["setup.py", "pyproject.toml", ".git"] |
project_root |
module | The root directory of the project. Relative paths are relative to this directory, unless specified otherwise. | Starting in the current test directory/module, scan up the directory hierarchy until one of the project_root_files are located. |
workflow_data_descriptor_file |
module | Path to the JSON file that describes the test data. | tests/test_data.json |
workflow_data_descriptors |
module | Mapping of workflow input/output names to values (as described in the Files section). | Loaded from the workflow_data_descriptor_file |
workflow_data_resolver |
module | Provides the DataResolver object that resolves test data; this should only need to be overridden for testing/debugging purposes |
DataResolver created from workflow_data_descriptors |
import_paths |
module | Provides the path to the file that lists the directories from which to import WDL dependencies | "import_paths.txt" |
import_dirs |
module | Provides the directories from which to import WDL dependencies | Loaded from import_paths file, if any |
Environment-specific configuration¶
There are several aspects of pytest-wdl that can be configured to the local environment, for example to enable the same tests to run both on a user’s development machine and in a continuous integration environment.
Environment-specific configuration is specified either or both of two places: a JSON configuration file and environment variables. Environment variables always take precendence over values in the configuration file. Keep in mind that (on a *nix system), environment variables can be set (semi-)permanently (using export
) or temporarily (using env
):
# Set environment variable durably
$ export FOO=bar
# Set environment variable only in the context of a single command
$ env FOO=bar echo "foo is $FOO"
Configuration file¶
The pytest-wdl configuration file is a JSON-format file. Its default location is $HOME/pytest_wdl_config.json
. Here is an example.
The available configuration options are listed in the following table:
configuration file key | environment variable | description | default | recommendation |
---|---|---|---|---|
cache_dir |
PYTEST_WDL_CACHE_DIR |
Directory to use for localizing test data files. | Temporary directory; a separate directory is used for each test module | pro: saves time when multiple tests rely on the same test data files; con: can cause conflicts, if tests use different files with the same name |
execution_dir |
PYTEST_WDL_EXECUTION_DIR |
Directory in which tests are executed | Temporary directory; a separate directory is used for each test function | Only use for debugging; use an absolute path |
proxies |
Configurable | Proxy server information; see details below | None | Use environment variable(s) to configure your proxy server(s), if any |
http_headers |
Configurable | HTTP header configuration that applies to all URLs matching a given pattern; see details below | None | Configure headers by URL pattern; configure headers for specific URLs in the test_data.json file |
show_progress |
N/A | Whether to show progress bars when downloading files | False | |
executors |
Executor-dependent | Configuration options specific to each executor; see below | None | |
N/A | LOGLEVEL |
Level of detail to log; can set to 'DEBUG', 'INFO', 'WARNING', or 'ERROR' | 'WARNING' | Use 'DEBUG' when developing plugins/fixtures/etc., otherwise 'WARNING' |
Proxies¶
In the proxies section of the configuration file, you can define the proxy servers for schemes used in data file URLs. The keys are scheme names and the values are either strings - environment variable names - or mappings with the following keys:
env
: The name of an environment variable in which to look for the proxy server address.value
: The value to use for the proxy server address, if the environment variable is not defined or is unset.
{
"proxies": {
"http": {
"env": "HTTP_PROXY"
},
"https": {
"value": "https://foo.com/proxy",
"env": "HTTPS_PROXY"
}
}
}
HTTP(S) Headers¶
In the http_headers section of the configuration file, you can define a list of headers to use when downloading data files. In addition to env
and value
keys (which are interpreted the same as for proxies, two additional keys are allowed:
name
: Required; the header namepattern
: A regular expression used to match the URL; if not specified, the header is used with all URLs.
{
"http_headers": [
{
"name": "X-JFrog-Art-Api",
"pattern": "http://my.company.com/artifactory/*",
"env": "TOKEN"
}
]
}
Executor-specific configuration¶
Cromwell¶
configuration file key | environment variable | description | default |
---|---|---|---|
java_bin |
JAVA_HOME |
Path to java executable; If not specified, then Java executable is expected to be in $JAVA_HOME/bin/java | None |
java_args |
JAVA_ARGS |
Arguments to add to the java command |
-Dconfig.file=<cromwell_config_file> (if cromwell_config_file is specified |
cromwell_jar |
CROMWELL_JAR |
Path to Cromwell JAR file | None |
N/A | CLASSPATH |
Java classpath; searched for a file matching "cromwell*.jar" if cromwell_jar is not specified |
None |
cromwell_config_file |
CROMWELL_CONFIG |
Path to Cromwell configuration file | None |
cromwell_args |
CROMWELL_ARGS |
Arguments to add to the cromwell run command |
None; recommended to use -Ddocker.hash-lookup.enabled=false to disable Docker lookup by hash |
Fixtures¶
There are two fixtures that control the loading of the user configuration:
fixture name | scope | description | default |
---|---|---|---|
user_config_file |
session | The location of the user configuration file | The value of the PYTEST_WDL_CONFIG environment variable if set, otherwise $HOME/pytest_wdl_config.json |
user_config |
session | Provides a UserConfiguration object that is used by other fixtures to access configuration values |
Default values are loaded from user_config_file , but most values can be overridden via environment variables (see below) |
Plugins¶
pytest-wdl provides the ability to implement 3rd-party plugins for data types and executors. When two plugins with the same name are present, the third-party plugin takes precedence over the built-in plugin (however, if there are two conflicting third-party plugins, an exception is raised).
Creating new data types¶
To create a new data type plugin, add a module in the data_types
package of pytest-wdl, or create it in your own 3rd party package.
Your plugin should subclass the pytest_wdl.core.DataFile
class and override its methods for _assert_contents_equal()
and/or _diff()
to define the behavior for this file type.
Next, add an entry point in setup.py. If the data type requires more dependencies to be installed, make sure to use a try/except ImportError
to warn about this and add the extra dependencies under the setup.py’s extras_require
. For example:
# plugin.py
try:
import mylib
except ImportError:
logger.warning(
"mytype is not available because the mylib library is not "
"installed"
)
setup(
...,
entry_points={
"pytest_wdl.data_types": [
"mydata = pytest_wdl.data_types.mytype:MyDataFile"
]
},
extras_require={
"mydata": ["mylib"]
}
)
In this example, the extra dependencies can be installed with pip install pytest-wdl[mydata]
.
Creating new executors¶
To create a new executor, add a module in the executors
package, or in your own 3rd party package.
Your plugin should subclass pytest_wdl.core.Executor
and implement the run_workflow()
method.
Next, add an entry point in setup.py. If the data type requires more dependencies to be installed, make sure to use a try/except ImportError
to warn about this and add the extra dependencies under the setup.py’s extras_require
(see example under Creating new data types). For example:
setup(
...,
entry_points={
"pytest_wdl.executors": [
"myexec = pytest_wdl.executors.myexec:MyExecutor"
]
},
extras_require={
"myexec": ["mylib"]
}
)