small fix on mock

tomoscan/esrf/mock.py

@@ -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):