diff --git a/scripts/ci/test_gate.sh b/scripts/ci/test_gate.sh index d6caf16..50dd94b 100755 --- a/scripts/ci/test_gate.sh +++ b/scripts/ci/test_gate.sh @@ -135,6 +135,10 @@ if cargo_toml_path.exists(): category_by_test_name[current_name] = category_for_path(current_path) def category_for_target(target: str) -> str: + for prefix in ('unittests ', 'doctests '): + if target.startswith(prefix): + target = target[len(prefix):] + break if target in category_by_path: return category_by_path[target] target_path = pathlib.PurePosixPath(target)