mirror of
https://github.com/DOI-DO/j40-cejst-2.git
synced 2025-02-25 02:54:19 -08:00
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")
|