96 lines
3.1 KiB
Python
96 lines
3.1 KiB
Python
import argparse
|
|
import os
|
|
import subprocess
|
|
from time import time
|
|
|
|
from test_utils import DirectoryExcursion
|
|
|
|
ROOT = os.path.normpath(
|
|
os.path.join(os.path.dirname(os.path.abspath(__file__)), os.path.pardir,
|
|
os.path.pardir))
|
|
r_package = os.path.join(ROOT, 'R-package')
|
|
|
|
|
|
def get_mingw_bin():
|
|
return os.path.join('c:/rtools40/mingw64/', 'bin')
|
|
|
|
|
|
def test_with_autotools(args):
|
|
with DirectoryExcursion(r_package):
|
|
mingw_bin = get_mingw_bin()
|
|
CXX = os.path.join(mingw_bin, 'g++.exe')
|
|
CC = os.path.join(mingw_bin, 'gcc.exe')
|
|
cmd = ['R.exe', 'CMD', 'INSTALL', str(os.path.curdir)]
|
|
env = os.environ.copy()
|
|
env.update({'CC': CC, 'CXX': CXX, "MAKE": "make -j$(nproc)"})
|
|
subprocess.check_call(cmd, env=env)
|
|
subprocess.check_call([
|
|
'R.exe', '-q', '-e',
|
|
"library(testthat); setwd('tests'); source('testthat.R')"
|
|
])
|
|
subprocess.check_call([
|
|
'R.exe', '-q', '-e',
|
|
"demo(runall, package = 'xgboost')"
|
|
])
|
|
|
|
|
|
def test_with_cmake(args):
|
|
os.mkdir('build')
|
|
with DirectoryExcursion('build'):
|
|
if args.compiler == 'mingw':
|
|
mingw_bin = get_mingw_bin()
|
|
CXX = os.path.join(mingw_bin, 'g++.exe')
|
|
CC = os.path.join(mingw_bin, 'gcc.exe')
|
|
env = os.environ.copy()
|
|
env.update({'CC': CC, 'CXX': CXX})
|
|
subprocess.check_call([
|
|
'cmake', os.path.pardir, '-DUSE_OPENMP=ON', '-DR_LIB=ON',
|
|
'-DCMAKE_CONFIGURATION_TYPES=Release', '-G', 'Unix Makefiles',
|
|
],
|
|
env=env)
|
|
subprocess.check_call(['make', '-j', 'install'])
|
|
elif args.compiler == 'msvc':
|
|
subprocess.check_call([
|
|
'cmake', os.path.pardir, '-DUSE_OPENMP=ON', '-DR_LIB=ON',
|
|
'-DCMAKE_CONFIGURATION_TYPES=Release', '-A', 'x64'
|
|
])
|
|
subprocess.check_call([
|
|
'cmake', '--build', os.path.curdir, '--target', 'install',
|
|
'--config', 'Release'
|
|
])
|
|
else:
|
|
raise ValueError('Wrong compiler')
|
|
with DirectoryExcursion(r_package):
|
|
subprocess.check_call([
|
|
'R.exe', '-q', '-e',
|
|
"library(testthat); setwd('tests'); source('testthat.R')"
|
|
])
|
|
subprocess.check_call([
|
|
'R.exe', '-q', '-e',
|
|
"demo(runall, package = 'xgboost')"
|
|
])
|
|
|
|
|
|
def main(args: argparse.Namespace) -> None:
|
|
start = time()
|
|
if args.build_tool == 'autotools':
|
|
test_with_autotools(args)
|
|
else:
|
|
test_with_cmake(args)
|
|
print("Duration:", time() - start)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument('--compiler',
|
|
type=str,
|
|
choices=['mingw', 'msvc'],
|
|
help='Compiler used for compiling CXX code.')
|
|
parser.add_argument(
|
|
'--build-tool',
|
|
type=str,
|
|
choices=['cmake', 'autotools'],
|
|
help='Build tool for compiling CXX code and install R package.')
|
|
args = parser.parse_args()
|
|
main(args)
|