Commit 6daa7b3e authored by payno's avatar payno Committed by payno
Browse files

small fix on mock

parent 80d43e3a
......@@ -92,26 +92,6 @@ class _ScanMock:
n_radio=n_radio, scan_range=scan_range, ref_n=ref_n, dark_n=dark_n
)
if n_ini_radio is None:
n_ini_radio = n_radio + n_extra_radio
call_acqui_end = True
else:
call_acqui_end = False
for i_radio in range(n_ini_radio):
self.add_radio(i_radio)
for i_extra_radio in range(n_extra_radio):
self.add_radio(n_radio + i_extra_radio)
for i_recons in range(n_recons):
self.add_reconstruction(i_recons)
for i_recons in range(n_pag_recons):
self.add_pag_reconstruction(i_recons)
if recons_vol is True:
self.add_recons_vol()
if call_acqui_end is True:
self.end_acquisition()
def add_radio(self, index=None):
raise NotImplementedError("Base class")
......@@ -233,6 +213,9 @@ class MockHDF5(_ScanMock):
self.add_initial_dark()
if create_ini_ref:
self.add_initial_ref()
if n_ini_proj is not None:
for i_radio in range(n_ini_proj):
self.add_radio(index=i_radio)
if create_final_ref:
self.add_final_ref()
self.scan = HDF5TomoScan(scan=self.scan_master_file, entry="entry")
......@@ -438,7 +421,7 @@ class MockEDF(_ScanMock):
for i_radio in range(n_ini_radio):
self.add_radio(i_radio)
for i_extra_radio in range(n_extra_radio):
self.add_radio(i_extra_radio)
self.add_radio(i_extra_radio + n_ini_radio)
for i_recons in range(n_recons):
self.add_reconstruction(i_recons)
for i_recons in range(n_pag_recons):
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment