From fcb2d62cc77e76747d38c82aa481f3ca6b1d0d02 Mon Sep 17 00:00:00 2001
From: Gregory Ashton <gregory.ashton@ligo.org>
Date: Mon, 3 Oct 2016 09:45:49 +0200
Subject: [PATCH] Fix tests to new sftfilepath interface

---
 tests.py | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/tests.py b/tests.py
index 1a9ccf6..b244368 100644
--- a/tests.py
+++ b/tests.py
@@ -92,8 +92,9 @@ class TestComputeFstat(Test):
         Writer.make_data()
         predicted_FS = Writer.predict_fstat()
 
-        search = pyfstat.ComputeFstat(tref=Writer.tref, sftlabel=Writer.label,
-                                      sftdir=Writer.outdir)
+        search = pyfstat.ComputeFstat(
+            tref=Writer.tref,
+            sftfilepath='{}/*{}*sft'.format(Writer.outdir, Writer.label))
         FS = search.run_computefstatistic_single_point(Writer.tstart,
                                                        Writer.tend,
                                                        Writer.F0,
@@ -119,8 +120,10 @@ class TestSemiCoherentGlitchSearch(Test):
         Writer.make_data()
 
         search = pyfstat.SemiCoherentGlitchSearch(
-            label=Writer.label, outdir=Writer.outdir, tref=Writer.tref,
-            tstart=Writer.tstart, tend=Writer.tend, nglitch=1)
+            label=self.label, outdir=outdir,
+            sftfilepath='{}/*{}*sft'.format(Writer.outdir, Writer.label),
+            tref=Writer.tref, tstart=Writer.tstart, tend=Writer.tend,
+            nglitch=1)
 
         FS = search.compute_nglitch_fstat(Writer.F0, Writer.F1, Writer.F2,
                                           Writer.Alpha, Writer.Delta,
@@ -177,7 +180,7 @@ class TestMCMCSearch(Test):
 
         search = pyfstat.MCMCSearch(
             label=self.label, outdir=outdir, theta_prior=theta, tref=tref,
-            sftlabel=self.label, sftdir=outdir,
+            sftfilepath='{}/*{}*sft'.format(Writer.outdir, Writer.label),
             tstart=tstart, tend=tend, nsteps=[100, 100], nwalkers=100,
             ntemps=1)
         search.run()
-- 
GitLab