Coverage for cuda/pathfinder/_utils/path_sort.py: 100.00%

12 statements  

« prev     ^ index     » next       coverage.py v7.16.0, created at 2026-09-03 02:41 +0000

1# SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved. 

2# SPDX-License-Identifier: Apache-2.0 

3 

4"""Deterministic path ordering helpers.""" 

5 

6from __future__ import annotations 

7 

8import os 

9import re 

10 

11_DIGIT_RUN = re.compile(r"(\d+)") 

12 

13 

14def numeric_aware_path_sort_key(path: str) -> tuple[tuple[int, int, str], ...]: 

15 """Return a key that compares embedded digit runs by numeric value.""" 

16 key: list[tuple[int, int, str]] = [] 1abcdefgkhi

17 for part in _DIGIT_RUN.split(os.path.normcase(path)): 1abcdefgkhi

18 if part.isdigit(): 1abcdefgkhi

19 normalized = part.lstrip("0") or "0" 1abcdefghi

20 key.append((1, len(normalized), normalized)) 1abcdefghi

21 else: 

22 key.append((0, 0, part)) 1abcdefgkhi

23 return tuple(key) 1abcdefgkhi