python-murfey/tests/util/test_init.py at 738c1872559365b17905fbff9612533740f7a0bb · DiamondLightSource/python-murfey · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
"""
Test module for the functions located in murfey.util.__init__
"""

from pathlib import Path

import pytest

from murfey.util import secure_path

secure_path_test_matrix = (
    # Keep spaces? | Input path | Expected output
    # Tomo workflow examples
    (
        False,
        "D:/User_22220202_090000_cm40000-1/Position_1_1_001_0.00_22220202_090000_EER.eer",
        "D:/User_22220202_090000_cm40000-1/Position_1_1_001_0.00_22220202_090000_EER.eer",
    ),
    (
        False,
        "D:/User_22220202_090000_cm40000-1_/Position_1_1_001_0.00_22220202_090000_EER.eer",
        "D:/User_22220202_090000_cm40000-1_/Position_1_1_001_0.00_22220202_090000_EER.eer",
    ),
    # CLEM workflow examples
    (
        True,
        "D:/Session/cm40000-1/images/My Sample/TileScan 1/Position 1--Stage00--Z00--C00.tif",
        "D:/Session/cm40000-1/images/My Sample/TileScan 1/Position 1--Stage00--Z00--C00.tif",
    ),
    (
        True,
        "D:/Session/cm40000-1/images/My Sample_/TileScan 1/Position 1--Stage00--Z00--C00.tif",
        "D:/Session/cm40000-1/images/My Sample_/TileScan 1/Position 1--Stage00--Z00--C00.tif",
    ),
    (
        True,
        "D:/Session/cm40000-1/images/My_Sample_/TileScan 1/Position 1--Stage00--Z00--C00.tif",
        "D:/Session/cm40000-1/images/My_Sample_/TileScan 1/Position 1--Stage00--Z00--C00.tif",
    ),
    # Go wild
    (
        True,
        "D:/some path/to_/this/repo!/my_file.txt",
        "D:/some path/to_/this/repo/my_file.txt",
    ),
    (
        True,
        "D:/some path__/to_/this/repo/my file.txt",
        "D:/some path__/to_/this/repo/my file.txt",
    ),
    (
        False,
        "D:/some path__/to_/this/repo/my file.txt",
        "D:/some_path__/to_/this/repo/my_file.txt",
    ),
)


@pytest.mark.parametrize("test_params", secure_path_test_matrix)
def test_secure_path(test_params: tuple[bool, str, str]):
    # Unpack test params
    keep_spaces, input_path, expected_output = test_params
    assert secure_path(Path(input_path), keep_spaces) == Path(expected_output)