def test_ha(self):
import tempfile
from mosbot import main
# Override the ephem.now() function to lie about the time
oldnow = ephem.now
ephem.now = bigben
tempdir = tempfile.mkdtemp()
fn1 = os.path.join(self.testdatadir, '2017-10-04-p3.json')
args = ['--script', os.path.join(tempdir, 'tonight.sh')]
args += [fn1, fn1, fn1]
mosbot = main(cmdlineargs=args, get_mosbot=True)
f = open(os.path.join(tempdir, 'seqnum.txt'), 'w')
f.write('1')
f.close()
mosbot.Nahead = 200
mosbot.update_for_image({
'airmass': 1.019,
'skybright': 19.372964228396995,
'transparency': 0.94911683705518535,
'zp': 26.348159194305978,
'seeing': 0.9776841626480679,
'band': 'z'
})
ephem.now = oldnow
def test_large_slew(self):
from mosbot import main
import tempfile
tempdir = tempfile.mkdtemp()
args = ['--script', os.path.join(tempdir, 'tonight.sh')]
args += self.jsonfiles
mosbot = main(cmdlineargs=args, get_mosbot=True)
mosbot.J1[0]['RA'] = 0
mosbot.J1[0]['dec'] = -90
# write sequence number
f = open(mosbot.seqnumpath, 'w')
f.write('5\n')
f.close()
fn = os.path.join(self.testdatadir, 'mos3.68488.im4.fits.fz')
#mosbot.process_file(fn)
mosbot.update_for_image({
'airmass': 1.019,
'skybright': 19.372964228396995,
'transparency': 0.94911683705518535,
'zp': 26.348159194305978,
'seeing': 0.9776841626480679,
'band': 'z'
})
def test_new_file(self):
from mosbot import main
import tempfile
tempdir = tempfile.mkdtemp()
args = ['--script', os.path.join(tempdir, 'tonight.sh')]
args.append('--adjust')
#args.append('--no-db')
args += self.jsonfiles
mosbot = main(cmdlineargs=args, get_mosbot=True)
# write sequence number
f = open(mosbot.seqnumpath, 'w')
f.write('10\n')
f.close()
fn = os.path.join(self.testdatadir, 'mos3.68488.im4.fits.fz')
mosbot.process_file(fn)
# Assert that entries have been added to the computed-exposure database
import obsdb
from obsdb.models import ComputedExptime, OtherPasses
c = ComputedExptime.objects.all()
for cc in c:
print(cc)
self.assertEqual(len(c), 19)
def test_no_cut_past(self):
from mosbot import main
import tempfile
import ephem
now = ephem.now()
print('Now:', now)
# Make the plans start an hour ago.
now -= (1. / 24.)
self.set_json_files(now)
tempdir = tempfile.mkdtemp()
args = [
'--script',
os.path.join(tempdir, 'tonight.sh'), '--no-cut-past', '--pass', '1'
]
args += self.jsonfiles
mosbot = main(cmdlineargs=args, get_mosbot=True)
# Touch forcepass1
f = open(os.path.join(tempdir, 'forcepass1'), 'w')
f.close()
tiles = [
650124, 650123, 652527, 654902, 652528, 654903, 657247, 650125,
657246, 657248
]
# BEFORE: check tile numbers
for i, tile in enumerate(tiles):
fn = os.path.join(tempdir, 'expose-%i.sh' % (i + 1))
txt = open(fn).read()
tilename = 'MzLS_%i_z' % tile
print('Checking for tile', tilename, 'in file', fn)
assert (tilename in txt)
# write sequence number
f = open(mosbot.seqnumpath, 'w')
f.write('5\n')
f.close()
mosbot.update_for_image({
'airmass': 1.019,
'skybright': 19.372964228396995,
'transparency': 0.94911683705518535,
'zp': 26.348159194305978,
'seeing': 0.9776841626480679,
'band': 'z'
})
# AFTER: check that tile numbers remain the same!
for i, tile in enumerate(tiles):
fn = os.path.join(tempdir, 'expose-%i.sh' % (i + 1))
txt = open(fn).read()
tilename = 'MzLS_%i_z' % tile
print('Checking for tile', tilename, 'in file', fn)
assert (tilename in txt)
def XXXtest_run(self):
from mosbot import main
import tempfile
tempdir = tempfile.mkdtemp()
args = ['--script', os.path.join(tempdir, 'tonight.sh')]
args += self.jsonfiles
mosbot = main(cmdlineargs=args, get_mosbot=True)
# write sequence number
f = open(mosbot.seqnumpath, 'w')
f.write('10\n')
f.close()
mosbot.run()
def test_new_file(self):
from mosbot import main
import tempfile
tempdir = tempfile.mkdtemp()
args = ['--script', os.path.join(tempdir, 'tonight.sh')]
args += self.jsonfiles
mosbot = main(cmdlineargs=args, get_mosbot=True)
# write sequence number
f = open(mosbot.seqnumpath, 'w')
f.write('10\n')
f.close()
fn = os.path.join(self.testdatadir, 'mos3.68488.im4.fits.fz')
mosbot.process_file(fn)
def XXXtest_run(self):
from mosbot import main
import tempfile
tempdir = tempfile.mkdtemp()
args = ['--script', os.path.join(tempdir, 'tonight.sh')]
args += self.jsonfiles
mosbot = main(cmdlineargs=args, get_mosbot=True)
# write sequence number
f = open(mosbot.seqnumpath, 'w')
f.write('10\n')
f.close()
mosbot.run()
def test_new_file(self):
from mosbot import main
import tempfile
tempdir = tempfile.mkdtemp()
args = ['--script', os.path.join(tempdir, 'tonight.sh')]
args += self.jsonfiles
mosbot = main(cmdlineargs=args, get_mosbot=True)
# write sequence number
f = open(mosbot.seqnumpath, 'w')
f.write('10\n')
f.close()
fn = os.path.join(self.testdatadir, 'mos3.68488.im4.fits.fz')
mosbot.process_file(fn)
def test_new_file_blank(self):
from mosbot import main
import tempfile
tempdir = tempfile.mkdtemp()
args = ['--script', os.path.join(tempdir, 'tonight.sh')]
args += self.jsonfiles
mosbot = main(cmdlineargs=args, get_mosbot=True)
mosbot.J1 = mosbot.J1[:10]
mosbot.J2 = mosbot.J2[:10]
mosbot.J3 = mosbot.J3[:10]
mosbot.Nahead = 15
# write sequence number
f = open(mosbot.seqnumpath, 'w')
f.write('5\n')
f.close()
fn = os.path.join(self.testdatadir, 'mos3.68488.im4.fits.fz')
#mosbot.process_file(fn)
mosbot.update_for_image({
'airmass': 1.019,
'extension': 'im4',
'pixscale': 0.26,
'camera': 'mosaic3',
'ra_ccd': 102.97109434817698,
'dec_ccd': 37.53083859605576,
'band': 'z',
'zp': 26.348159194305978,
'rawsky': 3522.2537,
'ndetected': 571,
'skybright': 19.372964228396995,
'dy': -50.904994833603581,
'transparency': 0.94911683705518535,
'seeing': 0.9776841626480679,
'dx': 9.0965935687085278,
'nmatched': 199
})
def test_sequence(self):
from mosbot import main
import tempfile
import ephem
now = ephem.now()
print('Now:', now)
# Make the plans start one hour ago.
now -= (1. / 24.)
self.set_json_files(now,
pass1='2017-03-01-p1.json',
pass2='2017-03-01-p2.json',
pass3='2017-03-01-p3.json')
tempdir = tempfile.mkdtemp()
args = [
'--script',
os.path.join(tempdir, 'tonight.sh'),
# '--no-cut-past',
'--sequence',
'--pass',
'1'
]
args += self.jsonfiles
mosbot = main(cmdlineargs=args, get_mosbot=True)
# Touch forcepass1
f = open(os.path.join(tempdir, 'forcepass1'), 'w')
f.close()
tiles = [686811]
# BEFORE: check tile numbers -- starting after now.
for i, tile in enumerate(tiles):
fn = os.path.join(tempdir, 'expose-%i.sh' % (i + 1))
txt = open(fn).read()
tilename = 'MzLS_%i_z' % tile
print('Checking for tile', tilename, 'in file', fn)
assert (tilename in txt)
print('Updating as though two hours late.')
mosbot.update_for_image(
{
'airmass': 1.019,
'skybright': 19.372964228396995,
'transparency': 0.94911683705518535,
'zp': 26.348159194305978,
'seeing': 0.9776841626480679,
'band': 'z'
},
now=now + (3. / 24.))
tiles = [664120, 661867, 661868, 659585, 659586, 661869]
# AFTER: check tile numbers.
for i, tile in enumerate(tiles):
fn = os.path.join(tempdir, 'expose-%i.sh' % (i + 1))
txt = open(fn).read()
tilename = 'MzLS_%i_z' % tile
print('Checking for tile', tilename, 'in file', fn)
assert (tilename in txt)
print('Updating after seqnum=3')
# Pretend that we're at seqnum=3 and assert that we proceed with the
# same set of tiles.
# write sequence number
f = open(mosbot.seqnumpath, 'w')
f.write('3\n')
f.close()
mosbot.update_for_image(
{
'airmass': 1.019,
'skybright': 19.372964228396995,
'transparency': 0.94911683705518535,
'zp': 26.348159194305978,
'seeing': 0.9776841626480679,
'band': 'z'
},
now=now + ((3. * 3600 + 3. * 80) / 86400.))
tiles = [664120, 661867, 661868, 659585, 659586, 661869]
# AFTER: check tile numbers.
for i, tile in enumerate(tiles):
fn = os.path.join(tempdir, 'expose-%i.sh' % (i + 1))
txt = open(fn).read()
tilename = 'MzLS_%i_z' % tile
print('Checking for tile', tilename, 'in file', fn)
assert (tilename in txt)