Merge pull request #82 from ReactiveBayes/81-select-test-files-using-make-test-test_args

Parallelize make test. #81 Add make test test_args= filter.

Author: bvdmitri

Project.toml

@@ -47,6 +47,7 @@ julia = "1.10"
 Aqua = "4c88cf16-eb10-579e-8560-4a9242c79595"
 BenchmarkTools = "6e4b80f9-dd63-53aa-95a3-0cdb28fa8baf"
 ForwardDiff = "f6369f11-7733-5829-9624-2563aa707210"
+Hwloc = "0e44f5e4-bd66-52a0-8798-143a42290a1d"
 JET = "c3a54625-cd67-489e-a8e7-0a5a0ff4e31b"
 Manifolds = "1cead3c2-87b3-11e9-0ccd-23c62b72b94e"
 Plots = "91a5bcdd-55d7-5caf-9e0b-520d859cae80"
@@ -57,4 +58,4 @@ StableRNGs = "860ef19b-820b-49d6-a774-d7a799459cd3"
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
 
 [targets]
-test = ["Test", "Aqua", "BenchmarkTools", "Plots", "Printf", "ForwardDiff", "Manifolds", "ReTestItems", "RollingFunctions", "JET", "StableRNGs"]
+test = ["Test", "Aqua", "BenchmarkTools", "Hwloc", "Plots", "Printf", "ForwardDiff", "Manifolds", "ReTestItems", "RollingFunctions", "JET", "StableRNGs"]

test/runtests.jl

@@ -1,6 +1,38 @@
-using ExponentialFamilyProjection, Test, ReTestItems, Random
+using Aqua, Hwloc, ReTestItems, ExponentialFamilyProjection, Random
 
 # Set the random seed for reproducibility of kl-divergence tests
 Random.seed!(42)
 
-runtests(ExponentialFamilyProjection; memory_threshold = 1.0)
+if get(ENV, "RUN_AQUA", "true") == "true"
+    Aqua.test_all(ExponentialFamilyProjection; ambiguities = false, piracies = false, deps_compat = (; check_extras = false, check_weakdeps = true))
+end
+
+nthreads, ncores = Hwloc.num_virtual_cores(), Hwloc.num_physical_cores()
+nthreads, ncores = max(nthreads, 1), max(ncores, 1)
+nworker_threads = Int(nthreads / ncores)
+memory_threshold = 1.0
+
+pkg_root = dirname(pathof(ExponentialFamilyProjection)) |> dirname
+test_root = joinpath(pkg_root, "test")
+
+if isempty(ARGS)
+    runtests(ExponentialFamilyProjection; nworkers = ncores, nworker_threads = nworker_threads, memory_threshold = memory_threshold)
+else
+    for arg in ARGS
+        # Translate colon syntax (e.g., rules:normal_mean_variance → rules/normal_mean_variance)
+        candidate = join(split(arg, ":"), "/")
+
+        # Build possible test paths relative to the package test directory
+        paths = [joinpath(test_root, candidate), joinpath(test_root, candidate * ".jl")]
+
+        path = findfirst(ispath, paths)
+
+        if path !== nothing
+            selected_path = paths[path]
+            @info "Running selective tests from $selected_path"
+            runtests(selected_path; nworkers = ncores, nworker_threads = nworker_threads, memory_threshold = memory_threshold)
+        else
+            @warn "Test target not found: $arg"
+        end
+    end
+end