mirror of
https://github.com/uutils/coreutils
synced 2024-12-14 07:12:44 +00:00
108 lines
3.3 KiB
Rust
108 lines
3.3 KiB
Rust
// spell-checker:ignore incorrectnumber
|
|
use crate::common::util::*;
|
|
|
|
#[test]
|
|
fn test_nproc() {
|
|
let nproc: u8 = new_ucmd!().succeeds().stdout_str().trim().parse().unwrap();
|
|
assert!(nproc > 0);
|
|
}
|
|
|
|
#[test]
|
|
fn test_nproc_all_omp() {
|
|
let result = new_ucmd!().arg("--all").succeeds();
|
|
|
|
let nproc: u8 = result.stdout_str().trim().parse().unwrap();
|
|
assert!(nproc > 0);
|
|
|
|
let result = TestScenario::new(util_name!())
|
|
.ucmd_keepenv()
|
|
.env("OMP_NUM_THREADS", "60")
|
|
.succeeds();
|
|
|
|
let nproc_omp: u8 = result.stdout_str().trim().parse().unwrap();
|
|
assert!(nproc_omp == 60);
|
|
|
|
let result = TestScenario::new(util_name!())
|
|
.ucmd_keepenv()
|
|
.env("OMP_NUM_THREADS", "1") // Has no effect
|
|
.arg("--all")
|
|
.succeeds();
|
|
let nproc_omp: u8 = result.stdout_str().trim().parse().unwrap();
|
|
assert!(nproc == nproc_omp);
|
|
|
|
// If the parsing fails, returns the number of CPU
|
|
let result = TestScenario::new(util_name!())
|
|
.ucmd_keepenv()
|
|
.env("OMP_NUM_THREADS", "incorrectnumber") // returns the number CPU
|
|
.succeeds();
|
|
let nproc_omp: u8 = result.stdout_str().trim().parse().unwrap();
|
|
assert!(nproc == nproc_omp);
|
|
}
|
|
|
|
#[test]
|
|
fn test_nproc_ignore() {
|
|
let result = new_ucmd!().succeeds();
|
|
let nproc_total: u8 = result.stdout_str().trim().parse().unwrap();
|
|
if nproc_total > 1 {
|
|
// Ignore all CPU but one
|
|
let result = TestScenario::new(util_name!())
|
|
.ucmd_keepenv()
|
|
.arg("--ignore")
|
|
.arg((nproc_total - 1).to_string())
|
|
.succeeds();
|
|
let nproc: u8 = result.stdout_str().trim().parse().unwrap();
|
|
assert!(nproc == 1);
|
|
// Ignore all CPU but one with a string
|
|
let result = TestScenario::new(util_name!())
|
|
.ucmd_keepenv()
|
|
.arg("--ignore= 1")
|
|
.succeeds();
|
|
let nproc: u8 = result.stdout_str().trim().parse().unwrap();
|
|
assert!(nproc_total - 1 == nproc);
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_nproc_ignore_all_omp() {
|
|
let result = TestScenario::new(util_name!())
|
|
.ucmd_keepenv()
|
|
.env("OMP_NUM_THREADS", "42")
|
|
.arg("--ignore=40")
|
|
.succeeds();
|
|
let nproc: u8 = result.stdout_str().trim().parse().unwrap();
|
|
assert!(nproc == 2);
|
|
}
|
|
|
|
#[test]
|
|
fn test_nproc_omp_limit() {
|
|
let result = TestScenario::new(util_name!())
|
|
.ucmd_keepenv()
|
|
.env("OMP_NUM_THREADS", "42")
|
|
.env("OMP_THREAD_LIMIT", "0")
|
|
.succeeds();
|
|
let nproc: u8 = result.stdout_str().trim().parse().unwrap();
|
|
assert!(nproc == 1);
|
|
|
|
let result = TestScenario::new(util_name!())
|
|
.ucmd_keepenv()
|
|
.env("OMP_NUM_THREADS", "42")
|
|
.env("OMP_THREAD_LIMIT", "2")
|
|
.succeeds();
|
|
let nproc: u8 = result.stdout_str().trim().parse().unwrap();
|
|
assert!(nproc == 2);
|
|
|
|
let result = TestScenario::new(util_name!())
|
|
.ucmd_keepenv()
|
|
.env("OMP_NUM_THREADS", "42")
|
|
.env("OMP_THREAD_LIMIT", "2bad")
|
|
.succeeds();
|
|
let nproc: u8 = result.stdout_str().trim().parse().unwrap();
|
|
assert!(nproc == 42);
|
|
|
|
let result = TestScenario::new(util_name!())
|
|
.ucmd_keepenv()
|
|
.env("OMP_THREAD_LIMIT", "1")
|
|
.succeeds();
|
|
let nproc: u8 = result.stdout_str().trim().parse().unwrap();
|
|
assert!(nproc == 1);
|
|
}
|