From fae7de84aef493668a351af44cbb76c04a6f56a7 Mon Sep 17 00:00:00 2001
From: Gregory Ashton <gregory.ashton@ligo.org>
Date: Thu, 15 Apr 2021 12:40:18 +0000
Subject: [PATCH] Reduce time required for the sampler tests
---
.gitlab-ci.yml | 1 -
test/integration/sample_from_the_prior_test.py | 5 +++--
test/integration/sampler_run_test.py | 9 +++++++--
3 files changed, 10 insertions(+), 5 deletions(-)
diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index add1efb9d..9da548155 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -115,7 +115,6 @@ python-3.7-samplers:
- python -m pip install .
- pytest test/integration/sampler_run_test.py --durations 10
- - pytest test/integration/sample_from_the_prior_test.py
# test samplers on python 3.6
python-3.6-samplers:
diff --git a/test/integration/sample_from_the_prior_test.py b/test/integration/sample_from_the_prior_test.py
index e48c3574d..64bb0bed1 100644
--- a/test/integration/sample_from_the_prior_test.py
+++ b/test/integration/sample_from_the_prior_test.py
@@ -90,10 +90,11 @@ class Test(unittest.TestCase):
likelihood=likelihood,
priors=priors,
sampler="dynesty",
- npoints=1000,
- walks=100,
+ nlive=1000,
+ nact=10,
outdir=self.outdir,
label=label,
+ save=False
)
pvalues = [
ks_2samp_wrapper(
diff --git a/test/integration/sampler_run_test.py b/test/integration/sampler_run_test.py
index 1e263986b..46f09bbb6 100644
--- a/test/integration/sampler_run_test.py
+++ b/test/integration/sampler_run_test.py
@@ -48,7 +48,12 @@ class TestRunningSamplers(unittest.TestCase):
likelihood=self.likelihood,
priors=self.priors,
sampler="dnest4",
- nlive=100,
+ max_num_levels=2,
+ num_steps=10,
+ new_level_interval=10,
+ num_per_step=10,
+ thread_steps=1,
+ num_particles=50,
save=False,
)
@@ -147,7 +152,7 @@ class TestRunningSamplers(unittest.TestCase):
sampler="pymc3",
draws=50,
tune=50,
- n_init=1000,
+ n_init=250,
save=False,
)
--
GitLab