mirror of
https://github.com/DOI-DO/j40-cejst-2.git
synced 2025-02-23 10:04:18 -08:00
* WIP on parallelizing * switching to get_tmp_path for nri * switching to get_tmp_path everywhere necessary * fixing linter errors * moving heavy ETLs to front of line * add hold * moving cdc places up * removing unnecessary print * moving h&t up * adding parallel to geo post * better census labels * switching to concurrent futures * fixing output
10 lines
366 B
Python
10 lines
366 B
Python
# pylint: disable=protected-access
|
|
import pytest
|
|
from data_pipeline.etl import constants, 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")
|