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)
