diff --git a/tests/test_ui_layer.py b/tests/test_ui_layer.py index fe2d30fdb..07b550366 100644 --- a/tests/test_ui_layer.py +++ b/tests/test_ui_layer.py @@ -12,8 +12,8 @@ # TODO: add files with invalid syntax -def test_can_read_cpu_pipeline(cpu_pipeline_gridrec): - pipline_stage_config = ui_layer.yaml_loader(cpu_pipeline_gridrec) +def test_can_read_cpu_pipeline(cpu_pipeline_gridrec: str): + pipline_stage_config = ui_layer.yaml_loader(Path(cpu_pipeline_gridrec)) assert len(pipline_stage_config) == 9 assert pipline_stage_config[0]["method"] == "standard_tomo" @@ -37,8 +37,8 @@ def test_can_read_cpu_pipeline(cpu_pipeline_gridrec): assert pipline_stage_config[8]["module_path"] == "httomolib.misc.images" -def test_can_read_gpu_pipeline(gpu_pipelineFBP): - pipline_stage_config = ui_layer.yaml_loader(gpu_pipelineFBP) +def test_can_read_gpu_pipeline(gpu_pipelineFBP: str): + pipline_stage_config = ui_layer.yaml_loader(Path(gpu_pipelineFBP)) assert len(pipline_stage_config) == 9 assert pipline_stage_config[0]["method"] == "standard_tomo" @@ -69,9 +69,9 @@ def test_uilayer_fails_with_nonexistant_file(file: str): UiLayer(Path(file), Path("doesnt_matter"), comm=comm) -def test_pipeline_build_no_loader(cpu_pipeline_gridrec, standard_data): +def test_pipeline_build_no_loader(cpu_pipeline_gridrec: str, standard_data: str): comm = MPI.COMM_NULL - LayerUI = UiLayer(cpu_pipeline_gridrec, standard_data, comm=comm) + LayerUI = UiLayer(Path(cpu_pipeline_gridrec), Path(standard_data), comm=comm) del LayerUI.PipelineStageConfig[0] with pytest.raises(ValueError) as e: LayerUI.build_pipeline() @@ -79,9 +79,9 @@ def test_pipeline_build_no_loader(cpu_pipeline_gridrec, standard_data): assert "no loader" in str(e) -def test_pipeline_build_duplicate_id(cpu_pipeline_gridrec, standard_data): +def test_pipeline_build_duplicate_id(cpu_pipeline_gridrec: str, standard_data: str): comm = MPI.COMM_NULL - LayerUI = UiLayer(cpu_pipeline_gridrec, standard_data, comm=comm) + LayerUI = UiLayer(Path(cpu_pipeline_gridrec), Path(standard_data), comm=comm) LayerUI.PipelineStageConfig[1]["id"] = "testid" LayerUI.PipelineStageConfig[2]["id"] = "testid" with pytest.raises(ValueError) as e: @@ -90,10 +90,10 @@ def test_pipeline_build_duplicate_id(cpu_pipeline_gridrec, standard_data): assert "duplicate id" in str(e) -def test_pipeline_build_cpu_pipeline(standard_data, cpu_pipeline_gridrec): +def test_pipeline_build_cpu_pipeline(standard_data: str, cpu_pipeline_gridrec: str): """Testing OutputRef.""" comm = MPI.COMM_WORLD - LayerUI = UiLayer(cpu_pipeline_gridrec, standard_data, comm=comm) + LayerUI = UiLayer(Path(cpu_pipeline_gridrec), Path(standard_data), comm=comm) pipeline = LayerUI.build_pipeline()