From 41ba9e8f00c89d72e5cb71c964722ce1ed4d8445 Mon Sep 17 00:00:00 2001 From: James Lamb Date: Tue, 3 Sep 2024 22:30:30 -0500 Subject: [PATCH] [ci] prevent C API tests from leaving files behind (fixes #6361) (#6637) --- tests/c_api_test/test_.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/tests/c_api_test/test_.py b/tests/c_api_test/test_.py index 743d59dd0..fdaa4e941 100644 --- a/tests/c_api_test/test_.py +++ b/tests/c_api_test/test_.py @@ -25,7 +25,7 @@ dtype_int64 = 3 def c_str(string): - return ctypes.c_char_p(string.encode("utf-8")) + return ctypes.c_char_p(str(string).encode("utf-8")) def load_from_file(filename, reference): @@ -203,8 +203,8 @@ def test_booster(tmp_path): LIB.LGBM_BoosterCreateFromModelfile(c_str(str(model_path)), ctypes.byref(num_total_model), ctypes.byref(booster2)) data = np.loadtxt(str(binary_example_dir / "binary.test"), dtype=np.float64) mat = data[:, 1:] - preb = np.empty(mat.shape[0], dtype=np.float64) - num_preb = ctypes.c_int64(0) + preds = np.empty(mat.shape[0], dtype=np.float64) + num_preds = ctypes.c_int64(0) data = np.asarray(mat.reshape(mat.size), dtype=np.float64) LIB.LGBM_BoosterPredictForMat( booster2, @@ -217,8 +217,8 @@ def test_booster(tmp_path): ctypes.c_int(0), ctypes.c_int(25), c_str(""), - ctypes.byref(num_preb), - preb.ctypes.data_as(ctypes.POINTER(ctypes.c_double)), + ctypes.byref(num_preds), + preds.ctypes.data_as(ctypes.POINTER(ctypes.c_double)), ) LIB.LGBM_BoosterPredictForFile( booster2, @@ -228,7 +228,7 @@ def test_booster(tmp_path): ctypes.c_int(0), ctypes.c_int(25), c_str(""), - c_str("preb.txt"), + c_str(tmp_path / "preds.txt"), ) LIB.LGBM_BoosterPredictForFile( booster2, @@ -238,7 +238,7 @@ def test_booster(tmp_path): ctypes.c_int(10), ctypes.c_int(25), c_str(""), - c_str("preb.txt"), + c_str(tmp_path / "preds.txt"), ) LIB.LGBM_BoosterFree(booster2)