mirror of
https://github.com/DOI-DO/j40-cejst-2.git
synced 2025-02-25 11:04:19 -08:00
6e6223cd5e
* Configure and run `black` and other pre-commit hooks Co-authored-by: matt bowen <matthew.r.bowen@omb.eop.gov>
11 lines
395 B
Python
11 lines
395 B
Python
# pylint: disable=protected-access
|
|
import pytest
|
|
from data_pipeline.etl import constants
|
|
from data_pipeline.etl import runner
|
|
|
|
|
|
def test_get_datasets_to_run():
|
|
assert runner._get_datasets_to_run(None) == constants.DATASET_LIST
|
|
assert runner._get_datasets_to_run("census") == [constants.CENSUS_INFO]
|
|
with pytest.raises(ValueError):
|
|
runner._get_datasets_to_run("doesnt_exist")
|