ornstein uhlenbeck test generates an output file to store spectrum

TurTLE/TEST.py

@@ -377,6 +377,11 @@ class TEST(_code):
                     ofile.require_group('tracers0')
                     for kk in ['position', 'velocity', 'vorticity', 'velocity_gradient']:
                         ofile['tracers0'].require_group(kk)
+            if (self.dns_type == 'ornstein_uhlenbeck_test'):
+                with h5py.File(os.path.join(self.work_dir, self.simname + '_ou_output.h5'), 'a') as ofile:
+                    ofile.require_group('spectra')
+                    ofile['spectra'].create_dataset('ou_spectra', (1,300,3,3),dtype=np.float)
+
         self.run(
                 nb_processes = opt.nb_processes,
                 nb_threads_per_process = opt.nb_threads_per_process,