def save_yaml_file(output_directory): yaml_filename = output_directory / "training.yml" yaml_section = [ { "cube_dir": str(output_directory / "cells"), "cell_def": "", "type": "cell", "signal_channel": 0, "bg_channel": 1, }, { "cube_dir": str(output_directory / "non_cells"), "cell_def": "", "type": "no_cell", "signal_channel": 0, "bg_channel": 1, }, ] yaml_contents = {"data": yaml_section} save_yaml(yaml_contents, yaml_filename)
def test_save_yaml(tmpdir): test_yaml_path = Path(tmpdir) / "test.yml" yaml.save_yaml(yaml_contents, test_yaml_path) test_yaml = yaml.open_yaml(test_yaml_path) saved_yaml = yaml.open_yaml(yaml_file) assert test_yaml == saved_yaml