def test_getdir3d_local_newplate_nocubes(self):
self._update_release('MPL-5')
config.forceDbOff()
errmsg = 'this is the end of the road. Try using some reasonable inputs.'
with self.assertRaises(MarvinError) as cm:
self._getdir3d('stack', mode='local', plateifu=self.new_plate)
self.assertIn(errmsg, str(cm.exception))
def test_getdir3d_remote_newplate_fail(self):
self._update_release('MPL-5')
config.forceDbOff()
errmsg = 'Could not retrieve a remote plate. If it is a mastar'
with self.assertRaises(MarvinError) as cm:
self._getdir3d('stack', mode='remote', plateifu=self.new_plate)
self.assertIn(errmsg, str(cm.exception))
def usedb(request):
''' fixture for optional turning off the db '''
if request.param:
config.forceDbOn()
else:
config.forceDbOff()
return config.db is not None
def test_mode_local_nodb(self):
config.forceDbOff()
self.mode = 'local'
errmsg = 'Query cannot be run in local mode'
with self.assertRaises(MarvinError) as cm:
self._set_modes(expmode='local')
self.assertIn(errmsg, str(cm.exception))
def test_convert_tool_auto_nodb(self):
self._set_remote()
config.forceDbOff()
r = self._run_query()
r.convertToTool('cube', mode='auto', limit=1)
self.assertEqual('remote', r.mode)
self.assertEqual('local', r.objects[0].mode)
self.assertEqual('file', r.objects[0].data_origin)
def fxnr(mode):
if mode == 'remote':
config.forceDbOff()
q = Query(search_filter='nsa.z < 0.1', release='MPL-4', mode=mode)
r = q.run()
yield r
q = None
r = None
if mode == 'remote':
config.forceDbOn()
def db(request):
"""Turn local db on or off.
Use this to parametrize over all db options.
"""
if travis and request.param not in travis.new_dbs:
pytest.skip('Skipping non-requested dbs')
if request.param == 'db':
config.forceDbOn()
else:
config.forceDbOff()
yield config.db is not None
config.forceDbOn()
def test_get_images_download_remote(self):
self._update_release('MPL-4')
config.forceDbOff()
self._get_cube()
localpath = self._make_paths(self.mangaredux,
mode='local',
inputs=[self.new_plateifu])
remotepath = self._make_paths(self.remoteredux,
mode='remote',
inputs=[self.new_plateifu])
self.assertFalse(os.path.isfile(localpath[0]))
image = getImagesByPlate(self.new_plate,
mode='remote',
as_url=True,
download=True)
self.assertTrue(os.path.isfile(localpath[0]))
self.assertIsNone(image)
def test_pickle_save_auto(self):
self._set_remote()
config.forceDbOff()
self._pickle_query(mode='auto', name='test_query.mpf')
def db_off():
"""Turn the DB off for a test, and reset it after."""
config.forceDbOff()
yield
config.forceDbOn()
def test_mode_auto_nodb(self):
config.forceDbOff()
self.set_sasurl(loc='local')
self.mode = 'auto'
self._set_modes(expmode='remote')
def db_off():
config.forceDbOff()
yield
config.forceDbOn()
# ymax [arcsec],
# image size [units of spaxel number])
spaxel_binid = fin['SPAXEL_BINID'].data
# Ngal x Nbin x (Mstar, Mstar_err)
mstar_all = fin['STELLAR_MASS_VORONOI'].data
# Select galaxy and binids
ind1 = np.where(plateifus == args.plateifu)[0][0]
ind_binid = spaxel_binid[ind1, :, :, 0].astype(int)
# Create 2D stellar mass array
mstar = np.ones(ind_binid.shape) * np.nan
for row, inds in enumerate(ind_binid):
ind_nans = np.where(inds == -99)
mstar[row] = mstar_all[ind1, inds, 0]
mstar[row][ind_nans] = np.nan
# trim mstar to match size of DAP maps and write to csv
config.forceDbOff()
cube = Cube(args.plateifu)
len_x = int(cube.header['NAXIS1'])
df = pd.DataFrame(mstar[:len_x, :len_x])
fout = join(path_data, 'manga-{}_mstar.csv'.format(args.plateifu))
df.to_csv(fout, index=False)
print('\nWrote:', fout)
fin.close()
def dboff():
config.forceDbOff()
yield True
config.forceDbOn()
def test_mma(self, mode):
if mode == 'remote':
config.forceDbOff()
q = Query()
assert q.mode == mode
def test_getdir3d_auto_plate_nodb(self):
self._update_release('MPL-5')
config.forceDbOff()
self._getdir3d('stack', mode='auto', plateifu=self.plate)
def test_getdir3d_auto_newplate(self):
self._update_release('MPL-4')
config.forceDbOff()
self._getdir3d('stack', mode='auto', plateifu=self.new_plate)
def test_getdir3d_mastar_auto(self):
self._update_release('MPL-5')
config.forceDbOff()
self._getdir3d('mastar', mode='auto', plateifu=self.mastar_plateifu)
