Mock Data Artifact
This module contains a mock version of the artifact manager for use with testing vivarium_public_health components.
- class vivarium_public_health.testing.mock_artifact.MockArtifactManager[source]
- property name
- load(entity_key, *args, **kwargs)[source]
Loads data associated with the given entity key.
- Parameters:
entity_key – The key associated with the expected data.
column_filters – Filters that subset the data by a categorical column and then remove the column from the raw data. They are supplied as keyword arguments to the load method in the form “column=value”.
- Returns:
The data associated with the given key, filtered down to the requested subset if the data is a dataframe.
- Return type:
Any