def test_players_in_crescent_order(qtbot):
"""
test if players are used in crescent order
"""
w = observation.Observation("/tmp")
#w.show()
qtbot.addWidget(w)
#w.mode = "new"
w.pj = config.EMPTY_PROJECT
w.ffmpeg_bin = "ffmpeg"
#w.leObservationId.setText("test")
media_file1 = "files/geese1.mp4"
w.check_media("1", media_file1, True)
#w.add_media_to_listview("1", media_file1)
media_file2 = "files/geese1.mp4"
w.check_media("1", media_file2, True)
w.twVideo1.cellWidget(1, 0).setCurrentIndex(2)
def handle_dialog():
qtbot.keyClick(w.qm, Qt.Key_Enter)
#qtbot.mouseClick(w.qm.Ok, Qt.LeftButton)
QTimer.singleShot(1000, handle_dialog)
qtbot.mouseClick(w.pbSave, Qt.LeftButton)
assert w.state == "refused"
def __init__(self, log, detArray, sky, scn, belv=0., nobs=1, elvDict=None):
self.log = log
self.detArray = detArray
self.sky = sky
self.scn = scn
self.belv = belv
self.nobs = nobs
self.elvDict = elvDict
#Store the elevation values and probabilities
if self.elvDict is not None:
self.elVals = np.array(elvDict.keys()).astype(np.float)
self.elFrac = np.array(elvDict.values()).astype(np.float)
else:
self.elVals = None
self.elFrac = None
#Store observation objects
self.observations = [
ob.Observation(self.log,
self,
self.detArray,
self.sky,
self.scn,
belv=self.belv) for n in range(nobs)
]
#Store sky temperatures and efficiencies
self.temps = np.array([obs.temp for obs in self.observations])
self.effics = np.array([obs.effic for obs in self.observations])
def test_imaka(psf_grid_raw):
time_start = time.time()
if psf_grid_raw is None:
psf_grid_raw = prepare_test()
h4rg = instrument.Instrument((6000,6000), 4.0, 10., 4.)
h4rg.scale = 0.1
sources = scene.Scene(stars_x, stars_y, stars_f)
psfgrid = psf.PSF_grid(psf_grid_raw)
print('Making Image: {0} sec'.format(time.time() - time_start))
obs = observation.Observation(h4rg, sources, psfgrid, 4, 3.0)
print('Saving Image: {0} sec'.format(time.time() - time_start))
obs.save_to_fits('tmp.fits', clobber=True)
# print 'Displaying Image'
# py.clf()
# py.imshow(obs.img, cmap='gist_heat')
# # Zoom in
# py.axis([2000, 3000, 2000, 3000])
# Print out some runtime information.
time_end = time.time()
run_time = time_end - time_start
print('Total Time: {0} seconds'.format(run_time))
print('Time Per Star: {0} seconds'.format(run_time / Nstars))
return obs
def test_ok(qtbot):
w = observation.Observation("/tmp")
#w.show()
qtbot.addWidget(w)
w.mode = "new"
w.pj = config.EMPTY_PROJECT
w.ffmpeg_bin = "ffmpeg"
media_file = "files/geese1.mp4"
w.leObservationId.setText("test")
w.check_media("1", media_file, True)
w.add_media_to_listview("1", media_file)
qtbot.mouseClick(w.pbSave, Qt.LeftButton)
assert w.state == "accepted"
assert w.pj == {
'time_format': 'hh:mm:ss',
'project_date': '',
'project_name': '',
'project_description': '',
'project_format_version': config.project_format_version,
'subjects_conf': {},
'behaviors_conf': {},
'observations': {},
'behavioral_categories': [],
'independent_variables': {},
'coding_map': {},
'behaviors_coding_map': [],
'converters': {}
}
def test_no_media_loaded(qtbot):
w = observation.Observation("/tmp")
w.show()
qtbot.addWidget(w)
qtbot.mouseClick(w.pbSave, Qt.LeftButton)
assert w.state == "refused"
def test_file_not_media(qtbot):
w = observation.Observation("/tmp")
w.show()
qtbot.addWidget(w)
w.mode = "new"
w.pj = config.EMPTY_PROJECT
w.ffmpeg_bin = "ffmpeg"
w.leObservationId.setText("test")
media_file = "files/test.boris"
assert w.check_media("1", media_file, True) == False
def test_no_obs_id(qtbot):
w = observation.Observation("/tmp")
w.show()
qtbot.addWidget(w)
w.ffmpeg_bin = "ffmpeg"
media_file = "files/geese1.mp4"
w.check_media("1", media_file, True)
w.add_media_to_listview("1", media_file)
qtbot.mouseClick(w.pbSave, Qt.LeftButton)
assert w.state == "refused"
def test_no_media_loaded(qtbot):
w = observation.Observation("/tmp")
#w.show()
qtbot.addWidget(w)
def handle_dialog():
qtbot.keyClick(w.qm, Qt.Key_Enter)
QTimer.singleShot(1000, handle_dialog)
qtbot.mouseClick(w.pbSave, Qt.LeftButton)
assert w.state == "refused"
def __init__(self, manual=False):
"""API for communicating with snakeoil.py and torcs. Launches torcs"""
print("Launching torcs...")
if manual:
self.__reset_torcs_manual()
else:
self.__reset_torcs()
print("Connecting to torcs...")
self.client = snakeoil.Client(p=3001)
self.client.maxSteps = np.inf
self.client.get_servers_input()
self.obs = observation.Observation()
self.obs.update_obs(self.client.S.d)
self.act = action.Action()
self.PI = 3.14159265359
def __init__(self, log, detArray, sky, scn, nobs=1):
self.log = log
self.detArray = detArray
self.sky = sky
self.scn = scn
#Store observation objects
self.observations = [
ob.Observation(self.log, self.detArray, self.sky, self.scn)
for n in range(nobs)
]
#Store sky temperatures and efficiencies
self.temps = np.array([obs.temp for obs in self.observations])
self.effics = np.array([obs.effic for obs in self.observations])
def test_file_not_media(qtbot):
"""
test if the loaded file is a media file or not
"""
w = observation.Observation("/tmp")
w.show()
qtbot.addWidget(w)
w.mode = "new"
w.pj = config.EMPTY_PROJECT
w.ffmpeg_bin = "ffmpeg"
w.leObservationId.setText("test")
media_file = "files/test.boris"
r, msg = w.check_media("1", media_file, True)
assert r == False
def test_cancel(qtbot):
w = observation.Observation("/tmp")
#w.show()
qtbot.addWidget(w)
w.mode = "new"
w.pj = config.EMPTY_PROJECT
w.ffmpeg_bin = "ffmpeg"
media_file = "files/geese1.mp4"
w.leObservationId.setText("test")
w.check_media("1", media_file, True)
w.add_media_to_listview("1", media_file)
qtbot.mouseClick(w.pbCancel, Qt.LeftButton)
assert w.pj == config.EMPTY_PROJECT
def test_no_obs_id(qtbot):
w = observation.Observation("/tmp")
#w.show()
qtbot.addWidget(w)
w.ffmpeg_bin = "ffmpeg"
media_file = "files/geese1.mp4"
w.check_media("1", media_file, True)
w.add_media_to_listview("1", media_file)
def handle_dialog():
qtbot.keyClick(w.qm, Qt.Key_Enter)
#qtbot.mouseClick(w.qm.Ok, Qt.LeftButton)
QTimer.singleShot(1000, handle_dialog)
qtbot.mouseClick(w.pbSave, Qt.LeftButton)
assert w.state == "refused"
def test_extract_wav_from_wav(qtbot):
w = observation.Observation("/tmp")
#w.show()
qtbot.addWidget(w)
w.mode = "new"
w.pj = config.EMPTY_PROJECT
w.ffmpeg_bin = "ffmpeg"
w.ffmpeg_cache_dir = "/tmp"
media_file = "files/test.wav"
w.leObservationId.setText("test")
w.check_media("1", media_file, True)
w.add_media_to_listview("1", media_file)
w.cbVisualizeSpectrogram.setChecked(True)
w.extract_wav()
assert os.path.isfile("/tmp/test.wav.wav")
import observation
NEW_LINE = '\n'
NUMBER_MESSAGES = 10000
with open('generated-data.csv', 'w') as output_file:
for _ in range(NUMBER_MESSAGES):
obs = observation.Observation()
for k in obs.oru['components'].keys():
hl7 = []
hl7.append(str(obs.msh_time.strftime("%Y%m%d%H%M%S")))
hl7.append(str(obs.req_time.strftime("%Y%m%d%H%M%S")))
hl7.append(str(obs.obs_end_time.strftime("%Y%m%d%H%M%S")))
hl7.append(str(k))
hl7.append(str(obs.oru['components'][k]['value']))
hl7.append(str(obs.oru['components'][k]['unit']))
hl7.append(str(obs.obx_time.strftime("%Y%m%d%H%M%S")))
hl7.append(str(obs.oru['method']))
output_file.write(','.join(hl7) + '\n')
