def test_default_and_overwrite_cases(self): script = [] expected_fits = commands.export_fits( script, self.testfile, out_dir=self.output_dir) # Manually ensure target does not pre-exist if os.path.isfile(expected_fits): shutil.rmtree(expected_fits) # Test in usual case, not-preexisting self.casa.run_script(script) self.assertTrue(os.path.isfile(expected_fits)) # Now it's there, check we overwrite OK (this is a no-brainer but still) script = [] expected_fits = commands.export_fits( script, self.testfile, out_dir=self.output_dir, overwrite=True) self.casa.run_script(script) # Now, attempt to run with overwriting turned off. This both ensures # we do not overwrite, and checks the error reporting mechanism: script = [] expected_fits = commands.export_fits( script, self.testfile, out_dir=self.output_dir, # overwrite=False #(By default) ) with self.assertRaises(RuntimeError): self.casa.run_script(script)
def test_default_and_overwrite_cases(self): script = [] expected_fits = commands.export_fits(script, self.testfile, out_dir=self.output_dir) # Manually ensure target does not pre-exist if os.path.isfile(expected_fits): shutil.rmtree(expected_fits) # Test in usual case, not-preexisting self.casa.run_script(script) self.assertTrue(os.path.isfile(expected_fits)) # Now it's there, check we overwrite OK (this is a no-brainer but still) script = [] expected_fits = commands.export_fits(script, self.testfile, out_dir=self.output_dir, overwrite=True) self.casa.run_script(script) # Now, attempt to run with overwriting turned off. This both ensures # we do not overwrite, and checks the error reporting mechanism: script = [] expected_fits = commands.export_fits( script, self.testfile, out_dir=self.output_dir, # overwrite=False #(By default) ) with self.assertRaises(RuntimeError): self.casa.run_script(script)
def test_specified_outpath(self): script = [] expected_fits = commands.export_fits( script, self.testfile, out_dir=None, out_path=self.output_dir + '/exportfitsout.fits') # Manually ensure target does not pre-exist if os.path.isdir(expected_fits): shutil.rmtree(expected_fits) # Test in usual case, not-preexisting casa_out, errs = self.casa.run_script(script) print '\n'.join(casa_out) self.assertTrue(os.path.isfile(expected_fits))
def test_specified_outpath(self): script = [] expected_fits = commands.export_fits(script, self.testfile, out_dir=None, out_path=self.output_dir + '/exportfitsout.fits') # Manually ensure target does not pre-exist if os.path.isdir(expected_fits): shutil.rmtree(expected_fits) # Test in usual case, not-preexisting casa_out, errs = self.casa.run_script(script) print '\n'.join(casa_out) self.assertTrue(os.path.isfile(expected_fits))
import drivecasa import drivecasa.commands as commands from drivecasa.utils import ensure_dir output_dir = './simulation_output' ensure_dir(output_dir) commands_logfile = os.path.join(output_dir, "./casa-clean-commands.log") output_visibility = os.path.join(output_dir, './foovis.ms') # output_visibility = os.path.join(output_dir, './vla-sim.MS') output_image = os.path.join(output_dir, './foo') if os.path.isfile(commands_logfile): os.unlink(commands_logfile) casa = drivecasa.Casapy(commands_logfile=commands_logfile, echo_to_stdout=True, ) script = [] clean_args = { "imsize": [512, 512], "cell": ['3.5arcsec'], } outmaps = commands.clean(script, output_visibility, niter=100, threshold_in_jy=1e-5, out_path=output_image, other_clean_args=clean_args, overwrite=True) commands.export_fits(script, outmaps.image, overwrite=True) casa.run_script(script)
output_dir = './simulation_output' ensure_dir(output_dir) commands_logfile = os.path.join(output_dir, "./casa-clean-commands.log") output_visibility = os.path.join(output_dir, './foovis.ms') # output_visibility = os.path.join(output_dir, './vla-sim.MS') output_image = os.path.join(output_dir, './foo') if os.path.isfile(commands_logfile): os.unlink(commands_logfile) casa = drivecasa.Casapy( commands_logfile=commands_logfile, echo_to_stdout=True, ) script = [] clean_args = { "imsize": [512, 512], "cell": ['3.5arcsec'], } outmaps = commands.clean(script, output_visibility, niter=100, threshold_in_jy=1e-5, out_path=output_image, other_clean_args=clean_args, overwrite=True) commands.export_fits(script, outmaps.image, overwrite=True) casa.run_script(script)