Source code for test.id03
from binoculars.backends import id03
import binoculars.util
import binoculars.space
import os
import numpy
import unittest
[docs]class TestCase(unittest.TestCase):
[docs] def setUp(self):
cfg_unparsed = {}
specfile = os.path.join(os.path.split(os.getcwd())[0], 'binoculars-binaries/examples/dataset/sixc_tutorial.spec' )
cfg_unparsed['specfile'] = specfile
cfg_unparsed['sdd'] = '1000'
cfg_unparsed['pixelsize'] = '0.055, 0.055'
cfg_unparsed['imagefolder'] = specfile.replace('sixc_tutorial.spec', 'images')
cfg_unparsed['centralpixel'] = '50 ,50'
numpy.save('mask.npy', numpy.identity(516))
cfg_unparsed['maskmatrix'] = 'mask.npy'
self.id03input = id03.EH2(cfg_unparsed)
self.projection = id03.HKLProjection({'resolution' : '0.01', 'limits' : '[0:, :-1, 0:0.2]'})
[docs] def test_IO(self):
jobs = list(self.id03input.generate_jobs(['820']))
destination_opts = self.id03input.get_destination_options(['820'])
imagedata = self.id03input.process_job(jobs[0])
intensity, weights, coords = imagedata.next()
projected = self.projection.project(*coords)
limits = self.projection.config.limits
space1 = binoculars.space.Space.from_image(self.projection.config.resolution, self.projection.get_axis_labels(), projected, intensity, weights, limits = limits[0])
print(space1)
intensity, weights, coords = imagedata.next()
projected = self.projection.project(*coords)
space2 = binoculars.space.Space.from_image(self.projection.config.resolution, self.projection.get_axis_labels(), projected, intensity, weights)
print(space1 + space2)
[docs] def tearDown(self):
os.remove('mask.npy')
if __name__ == '__main__':
unittest.main()