j40-cejst-2/data/data-pipeline/data_pipeline/tests/test_etl.py

# 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")