python-environment-tools/crates/pet-python-utils/tests/sys_prefix_test.rs at d517427cceaff7ceef140e66e4c8ea792511d052 · karthiknadig/python-environment-tools · 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
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
// Copyright (c) Microsoft Corporation.
// Licensed under the MIT License.
#![cfg(unix)]

mod common;
use pet_python_utils::version;
use std::path::PathBuf;

use common::resolve_test_path;

#[cfg(unix)]
#[test]
fn version_from_sys_prefix() {
    let path: PathBuf = resolve_test_path(&["unix", "pyvenv_cfg", ".venv"]);
    let version = version::from_prefix(&path).unwrap();
    assert_eq!(version, "3.12.1");

    let path: PathBuf = resolve_test_path(&["unix", "pyvenv_cfg", ".venv", "bin"]);
    let version = version::from_prefix(&path).unwrap();
    assert_eq!(version, "3.12.1");
}

#[cfg(unix)]
#[test]
fn version_from_sys_prefix_using_version_info_format() {
    let path: PathBuf = resolve_test_path(&["unix", "pyvenv_cfg", "hatch_env"]);
    let version = version::from_prefix(&path).unwrap();
    assert_eq!(version, "3.9.6.final.0");

    let path: PathBuf = resolve_test_path(&["unix", "pyvenv_cfg", "hatch_env", "bin"]);
    let version = version::from_prefix(&path).unwrap();
    assert_eq!(version, "3.9.6.final.0");
}

#[cfg(unix)]
#[test]
fn no_version_without_pyvenv_cfg_and_without_headers() {
    let path: PathBuf = resolve_test_path(&["unix", "pyvenv_cfg", "python3.9.9_without_headers"]);
    let version = version::from_prefix(&path);
    assert!(version.is_none());

    let path: PathBuf =
        resolve_test_path(&["unix", "pyvenv_cfg", "python3.9.9_without_headers", "bin"]);
    let version = version::from_prefix(&path);
    assert!(version.is_none());

    let path: PathBuf = resolve_test_path(&[
        "unix",
        "pyvenv_cfg",
        "python3.9.9_without_headers",
        "bin",
        "python",
    ]);
    let version = version::from_prefix(&path);
    assert!(version.is_none());
}

#[cfg(unix)]
#[test]
fn no_version_for_invalid_paths() {
    let path: PathBuf = resolve_test_path(&["unix_1234"]);
    let version = version::from_prefix(&path);
    assert!(version.is_none());
}

#[cfg(unix)]
#[test]
fn version_from_header_files() {
    let path: PathBuf = resolve_test_path(&["unix", "headers", "python3.9.9"]);
    let version = version::from_prefix(&path).unwrap();
    assert_eq!(version, "3.9.9");

    let path: PathBuf = resolve_test_path(&["unix", "headers", "python3.9.9", "bin"]);
    let version = version::from_prefix(&path).unwrap();
    assert_eq!(version, "3.9.9");

    let path: PathBuf = resolve_test_path(&["unix", "headers", "python3.10-dev", "bin"]);
    let version = version::from_prefix(&path).unwrap();
    assert_eq!(version, "3.10.14+");

    let path: PathBuf = resolve_test_path(&["unix", "headers", "python3.13", "bin"]);
    let version = version::from_prefix(&path).unwrap();
    assert_eq!(version, "3.13.0a5");
}