def test_monitor_accumulated(self): with tm.TimeMonitor("testtimer"): sleep(3) self.assertAlmostEqual(tm._monitors["testtimer"], 3, places=2) sleep(2) with tm.TimeMonitor("testtimer"): sleep(3) self.assertAlmostEqual(tm._monitors["testtimer"], 6, places=1)
def write(self, filename, a): with tm.TimeMonitor("MPI Write"): if self.comm.MyPID == 0: l.info("Writing " + filename) f = MPI.File.Open(MPI.COMM_WORLD, os.path.join(self.folder, filename), MPI.MODE_WRONLY | MPI.MODE_CREATE) f.Write_ordered(a) f.Close()
pass # write copy of config file to output folder with open(os.path.join(folder, "config.cfg"), "w") as config_output_file: config.write(config_output_file) # define data range i_from = 0 length = max_data_samples or len(h5py.File(input_filename, mode='r')['data']) # must be a multiple of baseline length length /= baseline_length length *= baseline_length comm.create_global_map("bas", length / baseline_length) # temperature and polarization done in 2 different runs for pol, comps in zip([False, True], ["T", "QU"]): with tm.TimeMonitor("Total"): with tm.TimeMonitor("Read input data"): # read data, pixels and baseline lengths in numpy arrays from hdf5 files pix, data, baseline_lengths = read_data( input_filename, i_from + comm.maps["bas"].MinMyGID() * baseline_length, i_from + (comm.maps["bas"].MaxMyGID() + 1) * baseline_length, nside, baseline_length, comm, pol=pol, maskdestripe=mask_filename) # replace the measured signal with a simulated signal if scan_gal_input_map: gal_input_map = gal2eq(hp.read_map(scan_gal_input_map, [0, 1, 2]))
def test_monitor_created(self): with tm.TimeMonitor("testtimer"): pass self.assertTrue(tm._monitors.has_key("testtimer"))
def test_summarize(self): with tm.TimeMonitor("testtimer"): sleep(3) printed = tm.summarize() self.assertGreater(printed.find("testtimer"), -1) self.assertGreater(printed.find("3"), -1)
def test_monitor_reset(self): with tm.TimeMonitor("testtimer"): sleep(3) tm.reset() self.assertEqual(tm._monitors["testtimer"], 0)
def test_monitor_incremented(self): with tm.TimeMonitor("testtimer"): sleep(3) self.assertAlmostEqual(tm._monitors["testtimer"], 3, places=2)