test_windows.ps1 5.71 KB
Newer Older
1
function Check-Output {
2
3
4
  param( [bool]$success )
  if (!$success) {
    $host.SetShouldExit(-1)
5
    exit 1
6
7
8
  }
}

Andrew Ziem's avatar
Andrew Ziem committed
9
# unify environment variable for Azure DevOps and AppVeyor
10
11
12
13
if (Test-Path env:APPVEYOR) {
  $env:APPVEYOR = "true"
}

14
15
16
17
18
if ($env:TASK -eq "r-package") {
  & $env:BUILD_SOURCESDIRECTORY\.ci\test_r_package_windows.ps1 ; Check-Output $?
  Exit 0
}

19
if ($env:TASK -eq "cpp-tests") {
20
21
22
  cmake -B build -S . -DBUILD_CPP_TEST=ON -DUSE_OPENMP=OFF -DUSE_DEBUG=ON -A x64
  cmake --build build --target testlightgbm --config Debug ; Check-Output $?
  .\Debug\testlightgbm.exe ; Check-Output $?
23
24
25
  Exit 0
}

26
if ($env:TASK -eq "swig") {
27
  $env:JAVA_HOME = $env:JAVA_HOME_8_X64  # there is pre-installed Eclipse Temurin 8 somewhere
28
29
30
31
  $ProgressPreference = "SilentlyContinue"  # progress bar bug extremely slows down download speed
  Invoke-WebRequest -Uri "https://github.com/microsoft/LightGBM/releases/download/v2.0.12/swigwin-4.0.2.zip" -OutFile $env:BUILD_SOURCESDIRECTORY/swig/swigwin.zip -UserAgent "NativeHost"
  Add-Type -AssemblyName System.IO.Compression.FileSystem
  [System.IO.Compression.ZipFile]::ExtractToDirectory("$env:BUILD_SOURCESDIRECTORY/swig/swigwin.zip", "$env:BUILD_SOURCESDIRECTORY/swig")
32
  $env:PATH = "$env:BUILD_SOURCESDIRECTORY/swig/swigwin-4.0.2;" + $env:PATH
33
34
  cmake -B build -S . -A x64 -DUSE_SWIG=ON ; Check-Output $?
  cmake --build build --target ALL_BUILD --config Release ; Check-Output $?
35
  if ($env:AZURE -eq "true") {
36
    cp $env:BUILD_SOURCESDIRECTORY/build/lightgbmlib.jar $env:BUILD_ARTIFACTSTAGINGDIRECTORY/lightgbmlib_win.jar ; Check-Output $?
37
38
39
40
  }
  Exit 0
}

41
42
43
44
# setup for Python
conda init powershell
conda activate
conda config --set always_yes yes --set changeps1 no
45
46
47
48
49
50

# ref:
# * https://stackoverflow.com/a/62897729/3986677
# * https://github.com/microsoft/LightGBM/issues/5899
conda install brotlipy

51
52
conda update -q -y conda
conda create -q -y -n $env:CONDA_ENV `
53
  cffi `
54
55
56
57
58
59
  cloudpickle `
  joblib `
  matplotlib `
  numpy `
  pandas `
  psutil `
60
  pyarrow `
61
62
63
64
65
66
67
68
69
  pytest `
  "python=$env:PYTHON_VERSION[build=*cpython]" `
  python-graphviz `
  scikit-learn `
  scipy ; Check-Output $?

if ($env:TASK -ne "bdist") {
  conda activate $env:CONDA_ENV
}
70

71
if ($env:TASK -eq "regular") {
72
73
  cmake -B build -S . -A x64 ; Check-Output $?
  cmake --build build --target ALL_BUILD --config Release ; Check-Output $?
74
75
  cd $env:BUILD_SOURCESDIRECTORY
  sh $env:BUILD_SOURCESDIRECTORY/build-python.sh install --precompile ; Check-Output $?
76
77
78
79
  cp $env:BUILD_SOURCESDIRECTORY/Release/lib_lightgbm.dll $env:BUILD_ARTIFACTSTAGINGDIRECTORY
  cp $env:BUILD_SOURCESDIRECTORY/Release/lightgbm.exe $env:BUILD_ARTIFACTSTAGINGDIRECTORY
}
elseif ($env:TASK -eq "sdist") {
80
81
82
  cd $env:BUILD_SOURCESDIRECTORY
  sh $env:BUILD_SOURCESDIRECTORY/build-python.sh sdist ; Check-Output $?
  sh $env:BUILD_SOURCESDIRECTORY/.ci/check_python_dists.sh $env:BUILD_SOURCESDIRECTORY/dist ; Check-Output $?
83
  cd dist; pip install @(Get-ChildItem *.gz) -v ; Check-Output $?
84
85
}
elseif ($env:TASK -eq "bdist") {
86
  # Import the Chocolatey profile module so that the RefreshEnv command
Andrew Ziem's avatar
Andrew Ziem committed
87
  # invoked below properly updates the current PowerShell session environment.
88
89
  $module = "$env:ChocolateyInstall\helpers\chocolateyProfile.psm1"
  Import-Module "$module" ; Check-Output $?
90
  RefreshEnv
91

92
93
94
95
  Write-Output "Current OpenCL drivers:"
  Get-ItemProperty -Path Registry::HKEY_LOCAL_MACHINE\SOFTWARE\Khronos\OpenCL\Vendors

  conda activate $env:CONDA_ENV
96
97
98
  cd $env:BUILD_SOURCESDIRECTORY
  sh "build-python.sh" bdist_wheel --integrated-opencl ; Check-Output $?
  sh $env:BUILD_SOURCESDIRECTORY/.ci/check_python_dists.sh $env:BUILD_SOURCESDIRECTORY/dist ; Check-Output $?
99
100
  cd dist; pip install --user @(Get-ChildItem *py3-none-win_amd64.whl) ; Check-Output $?
  cp @(Get-ChildItem *py3-none-win_amd64.whl) $env:BUILD_ARTIFACTSTAGINGDIRECTORY
101
} elseif (($env:APPVEYOR -eq "true") -and ($env:TASK -eq "python")) {
102
  cd $env:BUILD_SOURCESDIRECTORY
103
  if ($env:COMPILER -eq "MINGW") {
104
    sh $env:BUILD_SOURCESDIRECTORY/build-python.sh install --user --mingw ; Check-Output $?
105
  } else {
106
    sh $env:BUILD_SOURCESDIRECTORY/build-python.sh install --user; Check-Output $?
107
  }
108
109
}

110
if (($env:TASK -eq "sdist") -or (($env:APPVEYOR -eq "true") -and ($env:TASK -eq "python"))) {
111
  # cannot test C API with "sdist" task
112
  $tests = $env:BUILD_SOURCESDIRECTORY + "/tests/python_package_test"
113
} else {
114
  $tests = $env:BUILD_SOURCESDIRECTORY + "/tests"
115
116
}
if ($env:TASK -eq "bdist") {
117
118
  # Make sure we can do both CPU and GPU; see tests/python_package_test/test_dual.py
  $env:LIGHTGBM_TEST_DUAL_CPU_GPU = "1"
119
}
120

121
pytest $tests ; Check-Output $?
122

123
if (($env:TASK -eq "regular") -or (($env:APPVEYOR -eq "true") -and ($env:TASK -eq "python"))) {
124
125
  cd $env:BUILD_SOURCESDIRECTORY/examples/python-guide
  @("import matplotlib", "matplotlib.use('Agg')") + (Get-Content "plot_example.py") | Set-Content "plot_example.py"
126
  (Get-Content "plot_example.py").replace('graph.render(view=True)', 'graph.render(view=False)') | Set-Content "plot_example.py"  # prevent interactive window mode
127
  conda install -q -y -n $env:CONDA_ENV "h5py>3.0" ipywidgets notebook
128
  foreach ($file in @(Get-ChildItem *.py)) {
129
130
    @("import sys, warnings", "warnings.showwarning = lambda message, category, filename, lineno, file=None, line=None: sys.stdout.write(warnings.formatwarning(message, category, filename, lineno, line))") + (Get-Content $file) | Set-Content $file
    python $file ; Check-Output $?
131
  }  # run all examples
132
  cd $env:BUILD_SOURCESDIRECTORY/examples/python-guide/notebooks
133
  (Get-Content "interactive_plot_example.ipynb").replace('INTERACTIVE = False', 'assert False, \"Interactive mode disabled\"') | Set-Content "interactive_plot_example.ipynb"
134
  jupyter nbconvert --ExecutePreprocessor.timeout=180 --to notebook --execute --inplace *.ipynb ; Check-Output $?  # run all notebooks
135
}