test_multithreading.R 470 Bytes
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
test_that("getLGBMthreads() and setLGBMthreads() work as expected", {
    # works with integer input
    ret <- setLGBMthreads(2L)
    expect_null(ret)
    expect_equal(getLGBMthreads(), 2L)

    # works with float input
    ret <- setLGBMthreads(1.0)
    expect_null(ret)
    expect_equal(getLGBMthreads(), 1L)

    # setting to any negative number sets max threads to -1
    ret <- setLGBMthreads(-312L)
    expect_null(ret)
    expect_equal(getLGBMthreads(), -1L)
})