Windows exe ext.

dmlc · Apr 15, 2020 · 04f8d56 · 04f8d56
1 parent 9ed73f2
commit 04f8d56
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/tests/python/test_cli.py b/tests/python/test_cli.py
@@ -1,6 +1,7 @@
 import os
 import tempfile
 import unittest
+import platform
 import xgboost
 import subprocess
 import numpy
@@ -32,7 +33,12 @@ def test_cli_model(self):
             os.path.join(curdir, os.path.pardir, os.path.pardir))
         data_path = "{root}/demo/data/agaricus.txt.train?format=libsvm".format(
             root=project_root)
-        exe = os.path.join(project_root, 'xgboost')
+
+        if platform.system() == 'Windows':
+            exe = 'xgboost.exe'
+        else:
+            exe = 'xgboost'
+        exe = os.path.join(project_root, exe)
         assert os.path.exists(exe)
 
         with tempfile.TemporaryDirectory() as tmpdir: