def test_init_paths_P10(self): self.template_imagefile = "_master.h5" self.specfile_name = "anything" self.beamline = create_beamline("P10") params = { "root_folder": self.root_dir, "sample_name": self.sample_name, "scan_number": self.scan_number, "specfile_name": self.specfile_name, "template_imagefile": self.template_imagefile, } ( homedir, default_dirname, specfile, template_imagefile, ) = self.beamline.loader.init_paths(**params) self.assertEqual( homedir, self.root_dir + self.sample_name + "_{:05d}".format(self.scan_number) + "/", ) self.assertEqual(default_dirname, "e4m/") self.assertEqual(specfile, self.sample_name + "_{:05d}".format(self.scan_number)) self.assertEqual(template_imagefile, "S_00001_master.h5")
def test_init_paths_NANOMAX(self): self.template_imagefile = "%06d.h5" self.specfile_name = "anything" self.beamline = create_beamline("NANOMAX") params = { "root_folder": self.root_dir, "sample_name": self.sample_name, "scan_number": self.scan_number, "specfile_name": self.specfile_name, "template_imagefile": self.template_imagefile, } ( homedir, default_dirname, specfile, template_imagefile, ) = self.beamline.loader.init_paths(**params) self.assertEqual( homedir, self.root_dir + self.sample_name + "{:06d}".format(self.scan_number) + "/", ) self.assertEqual(default_dirname, "data/") self.assertEqual(specfile, None) self.assertEqual(template_imagefile, "%06d.h5")
def test_init_paths_specfile_P10_full_path(self): self.template_imagefile = "_master.h5" self.specfile_name = "anything" self.beamline = create_beamline("P10") self.setUpPyfakefs() valid_path = "/gpfs/bcdi/data" os.makedirs(valid_path) with open(valid_path + "/dummy.fio", "w") as f: f.write("dummy") params = { "root_folder": self.root_dir, "sample_name": self.sample_name, "scan_number": self.scan_number, "specfile_name": valid_path + "/dummy.fio", "template_imagefile": self.template_imagefile, } ( homedir, default_dirname, specfile, template_imagefile, ) = self.beamline.loader.init_paths(**params) self.assertEqual( homedir, self.root_dir + self.sample_name + "_{:05d}".format(self.scan_number) + "/", ) self.assertEqual(default_dirname, "e4m/") self.assertEqual(specfile, params["specfile_name"]) self.assertEqual(template_imagefile, "S_00001_master.h5")
def setUp(self): self.setUpPyfakefs() self.valid_path = "/gpfs/bcdi/data/" os.makedirs(self.valid_path) with open(self.valid_path + "defined.spec", "w") as f: f.write("test\n#UDETCALIB cen_pix_x=11.195,cen_pix_y=281.115," "pixperdeg=455.257," "det_distance_CC=1.434,det_distance_COM=1.193," "timestamp=2021-02-28T13:01:16.615422") with open(self.valid_path + "undefined.spec", "w") as f: f.write("test\n#this,is,bad") self.setup = Setup(beamline_name="ID01", detector_name="Maxipix") self.setup.detector.rootdir = self.valid_path self.beamline = create_beamline(name="ID01")
def setUp(self): self.conversion_table = labframe_to_xrayutil self.root_dir = "D:/data/test/" self.sample_name = "S" self.scan_number = 1 self.template_imagefile = None self.specfile_name = "test_spec" self.beamline = create_beamline("34ID") self.beam_direction = np.array([1, 0, 0]) self.offset_inplane = 1 self.params = { "wavelength": 1, "distance": 1, "pixel_x": 55000, "pixel_y": 55000, "inplane": 32, "outofplane": 28, "tilt": 0.005, "verbose": False, }
def setUp(self): self.conversion_table = labframe_to_xrayutil self.root_dir = "D:/data/SIXS/" self.sample_name = "S" self.scan_number = 1 self.specfile_name = self.root_dir + "alias_dict.txt" self.template_imagefile = "spare_ascan_mu_%05d.nxs" self.beamline = create_beamline("SIXS_2019") self.beam_direction = np.array([1, 0, 0]) self.offset_inplane = 1 self.params = { "wavelength": 1, "distance": 1, "pixel_x": 55000, "pixel_y": 55000, "inplane": 32, "outofplane": 28, "tilt": 0.005, "verbose": False, }
def setUp(self): self.conversion_table = labframe_to_xrayutil self.root_dir = "D:/data/P10/" self.sample_name = "S" self.scan_number = 1 self.template_imagefile = "_master.h5" self.specfile_name = "" self.beamline = create_beamline("P10") self.diffractometer = DiffractometerP10(sample_offsets=(0, 0, 0, 0)) self.beam_direction = np.array([1, 0, 0]) self.offset_inplane = 1 self.params = { "wavelength": 1, "distance": 1, "pixel_x": 55000, "pixel_y": 55000, "inplane": 32, "outofplane": 28, "tilt": 0.005, "verbose": False, }
def test_init_paths_ID01_BLISS(self): self.template_imagefile = "exp1_scan5.h5" self.specfile_name = None self.beamline = create_beamline("ID01BLISS") params = { "root_folder": self.root_dir, "sample_name": self.sample_name, "scan_number": self.scan_number, "specfile_name": self.specfile_name, "template_imagefile": self.template_imagefile, } ( homedir, default_dirname, specfile, template_imagefile, ) = self.beamline.loader.init_paths(**params) self.assertEqual(homedir, self.root_dir) self.assertEqual(default_dirname, "") self.assertEqual(specfile, None) self.assertEqual(template_imagefile, self.template_imagefile)
def test_init_paths_CRISTAL(self): self.template_imagefile = self.sample_name + "%d.nxs" self.specfile_name = "anything" self.beamline = create_beamline("CRISTAL") params = { "root_folder": self.root_dir, "sample_name": self.sample_name, "scan_number": self.scan_number, "specfile_name": self.specfile_name, "template_imagefile": self.template_imagefile, } ( homedir, default_dirname, specfile, template_imagefile, ) = self.beamline.loader.init_paths(**params) self.assertEqual( homedir, self.root_dir + self.sample_name + str(self.scan_number) + "/") self.assertEqual(default_dirname, "data/") self.assertEqual(specfile, None) self.assertEqual(template_imagefile, self.sample_name + "%d.nxs")
def test_init_paths_ID01(self): self.template_imagefile = "data_mpx4_%05d.edf.gz" self.specfile_name = "test" self.beamline = create_beamline("ID01") params = { "root_folder": self.root_dir, "sample_name": self.sample_name, "scan_number": self.scan_number, "specfile_name": self.specfile_name, "template_imagefile": self.template_imagefile, } ( homedir, default_dirname, specfile, template_imagefile, ) = self.beamline.loader.init_paths(**params) self.assertEqual( homedir, self.root_dir + self.sample_name + str(self.scan_number) + "/") self.assertEqual(default_dirname, "data/") self.assertEqual(specfile, "test") self.assertEqual(template_imagefile, "data_mpx4_%05d.edf.gz")
def test_init_paths_SIXS(self): self.template_imagefile = "spare_ascan_mu_%05d.nxs" self.specfile_name = self.root_dir + "alias_dict.txt" self.beamline = create_beamline("SIXS_2019") params = { "root_folder": self.root_dir, "sample_name": self.sample_name, "scan_number": self.scan_number, "specfile_name": self.specfile_name, "template_imagefile": self.template_imagefile, } ( homedir, default_dirname, specfile, template_imagefile, ) = self.beamline.loader.init_paths(**params) self.assertEqual( homedir, self.root_dir + self.sample_name + str(self.scan_number) + "/") self.assertEqual(default_dirname, "data/") self.assertEqual(specfile, self.specfile_name) self.assertEqual(template_imagefile, self.template_imagefile)
def test_find_outofplane_SIXS(self): beamline = create_beamline("SIXS_2019") self.assertTrue(beamline.find_outofplane() == 2)
def test_find_inplane_NANOMAX(self): beamline = create_beamline("NANOMAX") diffractometer = DiffractometerNANOMAX(sample_offsets=(0, 0)) self.assertTrue(beamline.find_inplane(diffractometer) == 0)
def test_find_outofplane_P10(self): beamline = create_beamline("P10") diffractometer = DiffractometerP10(sample_offsets=(0, 0, 0, 0)) self.assertTrue(beamline.find_outofplane(diffractometer) == 1)
def test_find_outofplane_SIXS(self): beamline = create_beamline("SIXS_2019") diffractometer = DiffractometerSIXS(sample_offsets=(0, 0)) self.assertTrue(beamline.find_outofplane(diffractometer) == 2)
def test_find_inplane_34ID(self): beamline = create_beamline("34ID") diffractometer = Diffractometer34ID(sample_offsets=(0, 90, 0)) self.assertTrue(beamline.find_inplane(diffractometer) == 0)
def test_find_outofplane_CRISTAL(self): beamline = create_beamline("CRISTAL") self.assertTrue(beamline.find_outofplane() == 1)
def test_repr(self): beamline = create_beamline("ID01") self.assertIsInstance(eval(repr(beamline)), BeamlineID01)
def test_find_outofplane_ID01(self): beamline = create_beamline("ID01") self.assertTrue(beamline.find_outofplane() == 1)
def test_find_inplane_ID01(self): beamline = create_beamline("ID01") diffractometer = DiffractometerID01(sample_offsets=(0, 0, 0)) self.assertTrue(beamline.find_inplane(diffractometer) == 0)
def test_find_outofplane_P10(self): beamline = create_beamline("P10") self.assertTrue(beamline.find_outofplane() == 1)
def test_find_outofplane_NANOMAX(self): beamline = create_beamline("NANOMAX") self.assertTrue(beamline.find_outofplane() == 1)
def test_find_outofplane_34ID(self): beamline = create_beamline("34ID") self.assertTrue(beamline.find_outofplane() == 1)
def test_find_inplane_CRISTAL(self): beamline = create_beamline("CRISTAL") diffractometer = DiffractometerCRISTAL(sample_offsets=(0, 0)) self.assertTrue(beamline.find_inplane(diffractometer) == 0)