def test_rt_migration_true(self): migrator = RtMigrator(self.wo) nsta = migrator.nsta ntr = len(self.obs_split[0]) ######################### # start loops ######################### # loop over segments (simulate real-time data) for itr in xrange(ntr): # update all input streams # loop over stations data_list = [] for ista in xrange(nsta): tr = self.obs_split[ista][itr] data_list.append(tr) # update data migrator.updateData(data_list) # update stacks migrator.updateStacks() # update max migrator.updateMax() ######################### # end loops ######################### # check we find the same absolute origin time #migrator.max_out.plot() st = migrator.max_out.stats.starttime + 45 ed = migrator.max_out.stats.starttime + 55 #max_out=migrator.max_out.slice(st,ed) #x_out=migrator.x_out.slice(st,ed) #y_out=migrator.y_out.slice(st,ed) #z_out=migrator.z_out.slice(st,ed) #plotMaxXYZ(max_out, x_out, y_out, z_out, 'test_out.png') max_trace = migrator.max_out.data tmax = np.argmax(max_trace) * self.dt tdiff = (migrator.max_out.stats.starttime + tmax) - (self.starttime + self.ot) self.assertEquals(tdiff, 0)
wo.opdict['dt'] = dt # split data files to simulate packets of real-time data obs_split = [] for obs in obs_list: split = obs / 3 obs_split.append(split) # generate ttimes_files for test #generate_random_test_points(wo, n_test, (x0, y0, z0)) generate_random_test_points(wo, n_test) tic = time.time() # set up migrator migrator = RtMigrator(wo) nsta = migrator.nsta ntr = len(obs_split[0]) ######################### # start loops ######################### # loop over segments (simulate real-time data) t_update_data = [] t_update_stacks = [] t_update_max = [] for itr in xrange(ntr): # update all input streams # loop over stations data_list = [] for ista in xrange(nsta):