test: DRY

Author: MilesCranmer

test/front_matter.jl

@@ -4,6 +4,8 @@ using TestItems
 using TestItemRunner
 using Preferences: set_preferences!
 
+ENV["SYMBOLIC_REGRESSION_IS_TESTING"] = "true"
+
 # Allow TEST_GROUP via ENV or the first CLI arg, default to "unit/basic"
 const TEST_GROUP = let g = get(ENV, "TEST_GROUP", nothing)
     g === nothing && length(ARGS) > 0 ? ARGS[1] : (g === nothing ? "unit/basic" : g)

test/runtests.jl

@@ -16,13 +16,10 @@ if startswith(TEST_GROUP, "integration/")
     Pkg.activate(integration_dir)
     Pkg.instantiate()
 
-    if startswith(integration_name, "ext/mlj")
-        ENV["SYMBOLIC_REGRESSION_IS_TESTING"] = "true"
-        if integration_name == "ext/mlj/templates"
-            include(joinpath(@__DIR__, "..", "example.jl"))
-            include(joinpath(@__DIR__, "..", "examples", "parameterized_function.jl"))
-            include(joinpath(@__DIR__, "..", "examples", "custom_types.jl"))
-        end
+    if startswith(integration_name, "ext/mlj") && integration_name == "ext/mlj/templates"
+        include(joinpath(@__DIR__, "..", "example.jl"))
+        include(joinpath(@__DIR__, "..", "examples", "parameterized_function.jl"))
+        include(joinpath(@__DIR__, "..", "examples", "custom_types.jl"))
     end
 
     @run_package_tests(

test/test_params.jl

@@ -4,8 +4,6 @@ using Optim: Optim
 using LineSearches: LineSearches
 using Test: Test
 
-ENV["SYMBOLIC_REGRESSION_IS_TESTING"] = "true"
-
 empty_all_globals!()
 
 const maximum_residual = 2e-2