def test_without_log_config_path(self, mock_read_config, mock_open): mock_read_config.return_value = {**self.config, "log_config": None} mock_open.side_effect = [ mock.mock_open(read_data=yaml.dump(self.log_config)).return_value ] init_logging() mock_read_config.assert_called_once() mock_open.assert_not_called()
def test_with_config(self, mock_read_config, mock_open): mock_read_config.return_value = self.config mock_open.side_effect = [ mock.mock_open(read_data=yaml.dump(self.log_config)).return_value ] init_logging(self.config) mock_read_config.assert_not_called() mock_open.assert_called_once_with("env/test/logging.yaml", "r", encoding="utf8")
import click from collector.config import read_config, init_logging from collector.tasks.merge_intensive_care_dataset.task import MergeIntensiveCareDataset from collector.store import LocalStore @click.command() @click.option("--name", help="The name of the dataset") @click.option("--input_folder", help="The folder containing the dataset files") @click.option("--output_folder", help="The folder where the dataset should be stored") def main(name, input_folder, output_folder): config = read_config() store = LocalStore(config["store"]["path"]) task = MergeIntensiveCareDataset(config["collector"], store) task(name=name, input_folder=input_folder, output_folder=output_folder) if __name__ == "__main__": # pylint: disable=no-value-for-parameter init_logging() main()