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)
