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')