def work_helper_image_files(): '''Test the static methods in scan_helper_image_files.''' helper = scan_helper_image_files() import dxtbx directory = os.path.join(os.path.split(dxtbx.__file__)[0], 'tests') template = 'image_###.dat' assert(len(scan_helper_image_files.template_directory_to_indices( template, directory)) == 20) assert(scan_helper_image_files.template_directory_index_to_image( template, directory, 1) == os.path.join(directory, 'image_001.dat')) assert(scan_helper_image_files.template_index_to_image( template, 1) == 'image_001.dat') assert(scan_helper_image_files.image_to_template_directory( os.path.join(directory, 'image_001.dat')) == (template, directory)) assert(scan_helper_image_files.image_to_index('image_001.dat') == 1) assert(scan_helper_image_files.image_to_template( 'image_001.dat') == 'image_###.dat') assert(scan_helper_image_files.image_to_index("image_6.8kev_1_001.cbf") == 1) return
def work_helper_image_files(): '''Test the static methods in scan_helper_image_files.''' helper = scan_helper_image_files() import dxtbx directory = os.path.join(os.path.split(dxtbx.__file__)[0], 'tests') template = 'image_###.dat' assert(len(scan_helper_image_files.template_directory_to_indices( template, directory)) == 20) assert(scan_helper_image_files.template_directory_index_to_image( template, directory, 1) == os.path.join(directory, 'image_001.dat')) assert(scan_helper_image_files.template_index_to_image( template, 1) == 'image_001.dat') assert(scan_helper_image_files.image_to_template_directory( os.path.join(directory, 'image_001.dat')) == (template, directory)) assert(scan_helper_image_files.image_to_index('image_001.dat') == 1) assert(scan_helper_image_files.image_to_template( 'image_001.dat') == 'image_###.dat') assert(scan_helper_image_files.image_to_index("image_6.8kev_1_001.cbf") == 1) return
def test_helper_image_files(): """Test the static methods in scan_helper_image_files.""" helper = scan_helper_image_files() import dxtbx directory = os.path.join(os.path.split(dxtbx.__file__)[0], "tests") template = "image_###.dat" assert (len( scan_helper_image_files.template_directory_to_indices( template, directory)) == 20) assert scan_helper_image_files.template_directory_index_to_image( template, directory, 1) == os.path.join(directory, "image_001.dat") assert (scan_helper_image_files.template_index_to_image( template, 1) == "image_001.dat") assert scan_helper_image_files.image_to_template_directory( os.path.join(directory, "image_001.dat")) == (template, directory) assert scan_helper_image_files.image_to_index("image_001.dat") == 1 assert scan_helper_image_files.image_to_template( "image_001.dat") == "image_###.dat" assert scan_helper_image_files.image_to_index( "image_6.8kev_1_001.cbf") == 1
def test_helper_image_files(image_test_dir): """Test the static methods in scan_helper_image_files.""" assert scan_helper_image_files() template = "image_###.dat" assert ( len( scan_helper_image_files.template_directory_to_indices( template, image_test_dir ) ) == 20 ) assert scan_helper_image_files.template_directory_index_to_image( template, image_test_dir, 1 ) == os.path.join(image_test_dir, "image_001.dat") assert ( scan_helper_image_files.template_index_to_image(template, 1) == "image_001.dat" ) assert scan_helper_image_files.image_to_template_directory( os.path.join(image_test_dir, "image_001.dat") ) == (template, image_test_dir) assert scan_helper_image_files.image_to_index("image_001.dat") == 1 assert scan_helper_image_files.image_to_template("image_001.dat") == "image_###.dat" assert scan_helper_image_files.image_to_index("image_6.8kev_1_001.cbf") == 1
def work_xScanFactory(): '''Test out the ScanFactory.''' import dxtbx directory = os.path.join(os.path.split(dxtbx.__file__)[0], 'tests') template = 'image_###.dat' xscans = [ ScanFactory.single( scan_helper_image_files.template_directory_index_to_image( template, directory, j + 1), scan_helper_image_formats.FORMAT_CBF, 1.0, 18 + 0.5 * j, 0.5, j) for j in range(20) ] xscans.reverse() try: print sum(xscans[1:], xscans[0]) print 'I should not see this message' except RuntimeError: pass xscans.sort() sum(xscans[1:], xscans[0]) a = ScanFactory.add(xscans[:10]) b = ScanFactory.add(xscans[10:]) a + b filename = scan_helper_image_files.template_directory_index_to_image( template, directory, 1) assert (len(ScanFactory.search(filename)) == 20) (a + b)[1:5] (a + b)[:10] cbf = os.path.join(directory, 'phi_scan_001.cbf') print 'OK'
def test_xScanFactory(): """Test out the ScanFactory.""" import dxtbx directory = os.path.join(os.path.split(dxtbx.__file__)[0], "tests") template = "image_###.dat" xscans = [ ScanFactory.single( scan_helper_image_files.template_directory_index_to_image( template, directory, j + 1), scan_helper_image_formats.FORMAT_CBF, 1.0, 18 + 0.5 * j, 0.5, j, ) for j in range(20) ] xscans.reverse() with pytest.raises(RuntimeError): print(sum(xscans[1:], xscans[0])) xscans.sort() sum(xscans[1:], xscans[0]) a = ScanFactory.add(xscans[:10]) b = ScanFactory.add(xscans[10:]) a + b filename = scan_helper_image_files.template_directory_index_to_image( template, directory, 1) assert len(ScanFactory.search(filename)) == 20 (a + b)[1:5] (a + b)[:10]
def test_xScanFactory(image_test_dir): """Test out the ScanFactory.""" template = "image_###.dat" xscans = [ ScanFactory.single_file( scan_helper_image_files.template_directory_index_to_image( template, image_test_dir, j + 1 ), 1.0, 18 + 0.5 * j, 0.5, j, ) for j in range(20) ] xscans.reverse() with pytest.raises(RuntimeError): print(sum(xscans[1:], xscans[0])) xscans.sort() sum(xscans[1:], xscans[0]) a = ScanFactory.add(xscans[:10]) b = ScanFactory.add(xscans[10:]) a + b filename = scan_helper_image_files.template_directory_index_to_image( template, image_test_dir, 1 ) assert len(ScanFactory.search(filename)) == 20 (a + b)[1:5] (a + b)[:10]
def test_single_deprecation(image_test_dir): """Test out the ScanFactory.""" with pytest.warns(DeprecationWarning): ScanFactory.single( scan_helper_image_files.template_directory_index_to_image( "image_###.dat", image_test_dir, 2 ), None, 1.0, 18 + 0.5, 0.5, 1, )
def search(filename): '''Get a list of files which appear to match the template and directory implied by the input filename. This could well be used to get a list of image headers to read and hence construct scans from.''' template, directory = \ scan_helper_image_files.image_to_template_directory(filename) indices = scan_helper_image_files.template_directory_to_indices( template, directory) return [scan_helper_image_files.template_directory_index_to_image( template, directory, index) for index in indices]
def work_xScanFactory(): '''Test out the ScanFactory.''' import dxtbx directory = os.path.join(os.path.split(dxtbx.__file__)[0], 'tests') template = 'image_###.dat' xscans = [ScanFactory.single( scan_helper_image_files.template_directory_index_to_image( template, directory, j + 1), scan_helper_image_formats.FORMAT_CBF, 1.0, 18 + 0.5 * j, 0.5, j) for j in range(20)] xscans.reverse() try: print sum(xscans[1:], xscans[0]) print 'I should not see this message' except RuntimeError, e: pass
def work_xscan_factory(): '''Test out the scan_factory.''' import dxtbx directory = os.path.join(os.path.split(dxtbx.__file__)[0], 'tests') template = 'image_###.dat' xscans = [scan_factory.single( scan_helper_image_files.template_directory_index_to_image( template, directory, j + 1), scan_helper_image_formats.FORMAT_CBF, 1.0, 18 + 0.5 * j, 0.5, j) for j in range(20)] xscans.reverse() try: print sum(xscans[1:], xscans[0]) print 'I should not see this message' except RuntimeError, e: pass
try: print sum(xscans[1:], xscans[0]) print 'I should not see this message' except RuntimeError, e: pass xscans.sort() sum(xscans[1:], xscans[0]) a = ScanFactory.add(xscans[:10]) b = ScanFactory.add(xscans[10:]) a + b filename = scan_helper_image_files.template_directory_index_to_image( template, directory, 1) assert(len(ScanFactory.search(filename)) == 20) (a + b)[1:5] (a + b)[:10] cbf = os.path.join(directory, 'phi_scan_001.cbf') print 'OK' def test_scan(): work_helper_image_files() work_helper_image_formats() work_xScanFactory()
try: print sum(xscans[1:], xscans[0]) print 'I should not see this message' except RuntimeError, e: pass xscans.sort() sum(xscans[1:], xscans[0]) a = scan_factory.add(xscans[:10]) b = scan_factory.add(xscans[10:]) a + b filename = scan_helper_image_files.template_directory_index_to_image( template, directory, 1) assert(len(scan_factory.search(filename)) == 20) (a + b)[1:5] (a + b)[:10] cbf = os.path.join(directory, 'phi_scan_001.cbf') print 'OK' def test_scan(): work_helper_image_files() work_helper_image_formats() work_xscan_factory()