Speed-up test by compiling NYC only once

siboehm · Nov 21, 2021 · fbf6f82 · fbf6f82
1 parent 3100582
commit fbf6f82
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 16 deletions.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -0,0 +1,23 @@
+import pytest
+from lightgbm import Booster
+
+from lleaves import Model
+
+
+@pytest.fixture(scope="session")
+def NYC_llvm():
+    llvm_model = Model(model_file="tests/models/NYC_taxi/model.txt")
+    llvm_model.compile()
+    return llvm_model
+
+
+@pytest.fixture(scope="session")
+def NYC_lgbm():
+    return Booster(model_file="tests/models/NYC_taxi/model.txt")
+
+
+@pytest.fixture(scope="session")
+def mtpl2_llvm():
+    llvm_model = Model(model_file="tests/models/mtpl2/model.txt")
+    llvm_model.compile()
+    return llvm_model
diff --git a/tests/test_parallel.py b/tests/test_parallel.py
@@ -1,22 +1,6 @@
 from ctypes import POINTER, c_double
 
 import numpy as np
-import pytest
-from lightgbm import Booster
-
-from lleaves import Model
-
-
-@pytest.fixture(scope="session")
-def NYC_llvm():
-    llvm_model = Model(model_file="tests/models/NYC_taxi/model.txt")
-    llvm_model.compile()
-    return llvm_model
-
-
-@pytest.fixture(scope="session")
-def NYC_lgbm():
-    return Booster(model_file="tests/models/NYC_taxi/model.txt")
 
 
 def test_parallel_edgecases(NYC_llvm, NYC_lgbm):