def setup_method(self):
self.diff = MockScannable()
self.wl = MockScannable()
self.wl.pos = 1.
self.eq = MockEquation()
self.scc = SimulatedCrystalCounter('det', self.diff, Fourc(), self.wl,
self.eq)
class TestFourCirclePlugin(object):
def setup_method(self):
self.geometry = Fourc()
def testGetName(self):
assert self.geometry.name == "fourc"
def testPhysicalAnglesToInternalPosition(self):
expected = self.geometry.physical_angles_to_internal_position(
(2, 4, 5, 6))
assert VliegPosition(0, 2, 0, 4, 5, 6) == expected
def testInternalPositionToPhysicalAngles(self):
result = self.geometry.internal_position_to_physical_angles(
VliegPosition(0, 2, 0, 4, 5, 6))
assert (norm(matrix([[2, 4, 5, 6]]) - matrix([list(result)])) < 0.001)
def testSupportsModeGroup(self):
assert self.geometry.supports_mode_group('fourc')
assert not self.geometry.supports_mode_group('fivecFixedAlpha')
assert not self.geometry.supports_mode_group('fivecFixedGamma')
def testGetFixedParameters(self):
self.geometry.fixed_parameters # check for exceptions
def testisParamaterFixed(self):
assert not self.geometry.parameter_fixed('made up parameter')
assert self.geometry.parameter_fixed('gamma')
assert self.geometry.parameter_fixed('alpha')
class TestFourCirclePlugin(unittest.TestCase):
def setUp(self):
self.geometry = Fourc()
def testGetName(self):
self.assertEqual(self.geometry.name, "fourc")
def testPhysicalAnglesToInternalPosition(self):
expected = self.geometry.physical_angles_to_internal_position((2, 4, 5, 6))
self.assert_(VliegPosition(0, 2, 0, 4, 5, 6) == expected)
def testInternalPositionToPhysicalAngles(self):
result = self.geometry.internal_position_to_physical_angles(
VliegPosition(0, 2, 0, 4, 5, 6))
self.assert_(norm(matrix([[2, 4, 5, 6]]) - matrix([list(result)]))
< 0.001)
def testSupportsModeGroup(self):
self.assertTrue(self.geometry.supports_mode_group('fourc'))
self.assertFalse(self.geometry.supports_mode_group('fivecFixedAlpha'))
self.assertFalse(self.geometry.supports_mode_group('fivecFixedGamma'))
def testGetFixedParameters(self):
self.geometry.fixed_parameters # check for exceptions
def testisParamaterFixed(self):
self.assertFalse(self.geometry.parameter_fixed('made up parameter'))
self.assertTrue(self.geometry.parameter_fixed('gamma'))
self.assertTrue(self.geometry.parameter_fixed('alpha'))
def createDiffcalcAndScannables(self):
self.en = DummyPD('en')
dummy = createDummyAxes(['delta', 'omega', 'chi', 'phi'])
scannableGroup = ScannableGroup('fourcgrp', dummy)
self.fourc = DiffractometerScannableGroup('fourc', None,
scannableGroup)
settings.hardware = ScannableHardwareAdapter(self.fourc, self.en)
settings.geometry = Fourc()
settings.ubcalc_persister = UbCalculationNonPersister()
from diffcalc.dc import dcvlieg as dc
reload(dc)
self.dc = dc
self.fourc.diffcalc = self.dc
self.hkl = Hkl('hkl', self.fourc, self.dc)
def setup_method(self):
self.geometry = Fourc()
def setUp(self):
self.geometry = Fourc()
