Rename tests/util.py to tests/fixtures.py for clarity (#851)

junpenglao · claude · web-flow · commit 08f5a428b303 · 2026-04-01T21:42:06.000+02:00
`util.py` was ambiguous — it looked like it might test `blackjax/util.py`
(which `test_util.py` actually does). `fixtures.py` signals shared test
infrastructure (base class, helpers) following pytest conventions.

Co-authored-by: Claude Sonnet 4.6 &lt;noreply@anthropic.com&gt;
diff --git a/tests/adaptation/test_low_rank_adaptation.py b/tests/adaptation/test_low_rank_adaptation.py
@@ -19,7 +19,7 @@
 
 import blackjax
 from blackjax.adaptation.low_rank_adaptation import _compute_low_rank_metric, _spd_mean
-from tests.util import BlackJAXTest
+from tests.fixtures import BlackJAXTest
 
 
 class SPDMeanTest(BlackJAXTest):
diff --git a/tests/adaptation/test_meads.py b/tests/adaptation/test_meads.py
@@ -7,7 +7,7 @@
 
 import blackjax
 from blackjax.adaptation.meads_adaptation import MEADSAdaptationState, base
-from tests.util import BlackJAXTest
+from tests.fixtures import BlackJAXTest
 
 
 def make_logdensity(dim=2):
diff --git a/tests/fixtures.py b/tests/fixtures.py
diff --git a/tests/mcmc/test_elliptical_slice.py b/tests/mcmc/test_elliptical_slice.py
@@ -26,7 +26,7 @@
     ellipsis,
     init,
 )
-from tests.util import BlackJAXTest, std_normal_logdensity
+from tests.fixtures import BlackJAXTest, std_normal_logdensity
 
 # ---------------------------------------------------------------------------
 # ellipsis geometry
diff --git a/tests/mcmc/test_mala.py b/tests/mcmc/test_mala.py
@@ -21,7 +21,7 @@
 
 import blackjax.mcmc.diffusions as diffusions
 import blackjax.mcmc.mala as mala
-from tests.util import BlackJAXTest, std_normal_logdensity
+from tests.fixtures import BlackJAXTest, std_normal_logdensity
 
 # ---------------------------------------------------------------------------
 # mcmc/diffusions.overdamped_langevin
diff --git a/tests/sgmcmc/test_diffusions.py b/tests/sgmcmc/test_diffusions.py
@@ -20,7 +20,7 @@
 from absl.testing import absltest, parameterized
 
 import blackjax.sgmcmc.diffusions as diffusions
-from tests.util import BlackJAXTest
+from tests.fixtures import BlackJAXTest
 
 
 class OverdampedLangevinTest(BlackJAXTest):
diff --git a/tests/sgmcmc/test_kernels.py b/tests/sgmcmc/test_kernels.py
@@ -23,7 +23,7 @@
 import blackjax.sgmcmc.sghmc as sghmc
 import blackjax.sgmcmc.sgld as sgld
 import blackjax.sgmcmc.sgnht as sgnht
-from tests.util import BlackJAXTest
+from tests.fixtures import BlackJAXTest
 
 # ---------------------------------------------------------------------------
 # Shared helpers
diff --git a/tests/smc/__init__.py b/tests/smc/__init__.py
@@ -2,7 +2,7 @@
 import jax.scipy.stats as stats
 import numpy as np
 
-from tests.util import BlackJAXTest
+from tests.fixtures import BlackJAXTest
 
 
 class SMCLinearRegressionTestCase(BlackJAXTest):
diff --git a/tests/smc/test_from_mcmc.py b/tests/smc/test_from_mcmc.py
@@ -22,7 +22,7 @@
 import blackjax.smc.resampling as resampling
 from blackjax.smc.base import SMCState, init
 from blackjax.smc.from_mcmc import build_kernel, unshared_parameters_and_step_fn
-from tests.util import BlackJAXTest, std_normal_logdensity
+from tests.fixtures import BlackJAXTest, std_normal_logdensity
 
 # ---------------------------------------------------------------------------
 # unshared_parameters_and_step_fn
diff --git a/tests/smc/test_inner_kernel_tuning.py b/tests/smc/test_inner_kernel_tuning.py
@@ -24,9 +24,9 @@
     particles_means,
     particles_stds,
 )
+from tests.fixtures import BlackJAXTest
 from tests.mcmc.test_sampling import irmh_proposal_distribution
 from tests.smc import SMCLinearRegressionTestCase
-from tests.util import BlackJAXTest
 
 
 class MultivariableParticlesDistribution:
diff --git a/tests/smc/test_kernel_compatibility.py b/tests/smc/test_kernel_compatibility.py
@@ -6,7 +6,7 @@
 from blackjax import adaptive_tempered_smc
 from blackjax.mcmc.random_walk import normal
 from blackjax.smc import extend_params
-from tests.util import BlackJAXTest
+from tests.fixtures import BlackJAXTest
 
 
 class SMCAndMCMCIntegrationTest(BlackJAXTest):
diff --git a/tests/smc/test_pretuning.py b/tests/smc/test_pretuning.py
@@ -14,8 +14,8 @@
     init,
     update_parameter_distribution,
 )
+from tests.fixtures import BlackJAXTest
 from tests.smc import SMCLinearRegressionTestCase
-from tests.util import BlackJAXTest
 
 
 class TestMeasureOfChainMixing(unittest.TestCase):
diff --git a/tests/smc/test_smc.py b/tests/smc/test_smc.py
@@ -13,7 +13,7 @@
 from blackjax.smc.base import extend_params, init, step
 from blackjax.smc.tempered import update_and_take_last
 from blackjax.smc.waste_free import update_waste_free
-from tests.util import BlackJAXTest
+from tests.fixtures import BlackJAXTest
 
 
 def logdensity_fn(position):
diff --git a/tests/smc/test_tempered_smc.py b/tests/smc/test_tempered_smc.py
@@ -14,8 +14,8 @@
 import blackjax.smc.solver as solver
 from blackjax import adaptive_tempered_smc, tempered_smc
 from blackjax.smc import extend_params
+from tests.fixtures import BlackJAXTest
 from tests.smc import SMCLinearRegressionTestCase
-from tests.util import BlackJAXTest
 
 
 def inference_loop(kernel, rng_key, initial_state):
diff --git a/tests/smc/test_waste_free_smc.py b/tests/smc/test_waste_free_smc.py
@@ -14,9 +14,9 @@
 from blackjax import adaptive_tempered_smc, tempered_smc
 from blackjax.smc import extend_params
 from blackjax.smc.waste_free import update_waste_free, waste_free_smc
+from tests.fixtures import BlackJAXTest
 from tests.smc import SMCLinearRegressionTestCase
 from tests.smc.test_tempered_smc import inference_loop
-from tests.util import BlackJAXTest
 
 
 class WasteFreeSMCTest(SMCLinearRegressionTestCase):
diff --git a/tests/vi/test_fullrank_vi.py b/tests/vi/test_fullrank_vi.py
@@ -26,7 +26,7 @@
     sample,
     step,
 )
-from tests.util import BlackJAXTest, std_normal_logdensity
+from tests.fixtures import BlackJAXTest, std_normal_logdensity
 
 
 class FRVIUnitTest(BlackJAXTest):
diff --git a/tests/vi/test_meanfield_vi.py b/tests/vi/test_meanfield_vi.py
@@ -14,7 +14,7 @@
     sample,
     step,
 )
-from tests.util import BlackJAXTest, std_normal_logdensity
+from tests.fixtures import BlackJAXTest, std_normal_logdensity
 
 
 class MFVIUnitTest(BlackJAXTest):
diff --git a/tests/vi/test_multipathfinder.py b/tests/vi/test_multipathfinder.py
@@ -7,7 +7,7 @@
 from absl.testing import absltest
 
 import blackjax
-from tests.util import BlackJAXTest, std_normal_logdensity
+from tests.fixtures import BlackJAXTest, std_normal_logdensity
 
 
 class MultipathfinderTest(BlackJAXTest):
diff --git a/tests/vi/test_pathfinder.py b/tests/vi/test_pathfinder.py
@@ -19,7 +19,7 @@
 from absl.testing import absltest
 
 from blackjax.vi.pathfinder import PathfinderState, approximate, sample
-from tests.util import BlackJAXTest, std_normal_logdensity
+from tests.fixtures import BlackJAXTest, std_normal_logdensity
 
 
 class PathfinderApproximateTest(BlackJAXTest):
diff --git a/tests/vi/test_schrodinger_follmer.py b/tests/vi/test_schrodinger_follmer.py
@@ -7,7 +7,7 @@
 from absl.testing import absltest
 
 from blackjax.vi.schrodinger_follmer import as_top_level_api as schrodinger_follmer
-from tests.util import BlackJAXTest
+from tests.fixtures import BlackJAXTest
 
 
 class SchrodingerFollmerTest(BlackJAXTest):
diff --git a/tests/vi/test_svgd.py b/tests/vi/test_svgd.py
@@ -16,7 +16,7 @@
     rbf_kernel,
     update_median_heuristic,
 )
-from tests.util import BlackJAXTest
+from tests.fixtures import BlackJAXTest
 
 
 class SVGDUnitTest(BlackJAXTest):

Original file line number	Diff line number	Diff line change
`@@ -26,7 +26,7 @@`
`26`	`26`	`ellipsis,`
`27`	`27`	`init,`
`28`	`28`	`)`
`29`		`-from tests.util import BlackJAXTest, std_normal_logdensity`
	`29`	`+from tests.fixtures import BlackJAXTest, std_normal_logdensity`
`30`	`30`
`31`	`31`	`# ---------------------------------------------------------------------------`
`32`	`32`	`# ellipsis geometry`
Original file line number	Diff line number	Diff line change
`@@ -24,9 +24,9 @@`
`24`	`24`	`particles_means,`
`25`	`25`	`particles_stds,`
`26`	`26`	`)`
	`27`	`+from tests.fixtures import BlackJAXTest`
`27`	`28`	`from tests.mcmc.test_sampling import irmh_proposal_distribution`
`28`	`29`	`from tests.smc import SMCLinearRegressionTestCase`
`29`		`-from tests.util import BlackJAXTest`
`30`	`30`
`31`	`31`
`32`	`32`	`class MultivariableParticlesDistribution:`