class TestParameterManager(object):
def setup_method(self):
self.hw = createMockHardwareMonitor()
self.ms = ModeSelector(createMockDiffractometerGeometry())
self.pm = VliegParameterManager(createMockDiffractometerGeometry(),
self.hw, self.ms)
def testDefaultParameterValues(self):
assert self.pm.get_constraint('alpha') == 0
assert self.pm.get_constraint('gamma') == 0
with pytest.raises(DiffcalcException):
self.pm.get_constraint('not-a-parameter-name')
def testSetParameter(self):
self.pm.set_constraint('alpha', 10.1)
assert self.pm.get_constraint('alpha') == 10.1
def testSetTrackParameter_isParameterChecked(self):
assert not self.pm.isParameterTracked('alpha')
self.pm.set_constraint('alpha', 9)
self.pm.setTrackParameter('alpha', True)
assert self.pm.isParameterTracked('alpha') == True
with pytest.raises(DiffcalcException):
self.pm.set_constraint('alpha', 10)
self.hw.get_position.return_value = 888, 11, 999
assert self.pm.get_constraint('alpha') == 11
print self.pm.reportAllParameters()
print "**"
print self.ms.reportCurrentMode()
print self.pm.reportParametersUsedInCurrentMode()
self.pm.setTrackParameter('alpha', False)
assert not self.pm.isParameterTracked('alpha')
assert self.pm.get_constraint('alpha') == 11
self.hw.get_position.return_value = 888, 12, 999
assert self.pm.get_constraint('alpha') == 11
self.pm.set_constraint('alpha', 13)
assert self.pm.get_constraint('alpha') == 13
class TestParameterManager(unittest.TestCase):
def setUp(self):
self.hw = createMockHardwareMonitor()
self.ms = ModeSelector(createMockDiffractometerGeometry())
self.pm = VliegParameterManager(createMockDiffractometerGeometry(),
self.hw, self.ms)
def testDefaultParameterValues(self):
self.assertEqual(self.pm.get_constraint('alpha'), 0)
self.assertEqual(self.pm.get_constraint('gamma'), 0)
self.assertRaises(DiffcalcException, self.pm.get_constraint,
'not-a-parameter-name')
def testSetParameter(self):
self.pm.set_constraint('alpha', 10.1)
self.assertEqual(self.pm.get_constraint('alpha'), 10.1)
def testSetTrackParameter_isParameterChecked(self):
self.assertEqual(self.pm.isParameterTracked('alpha'), False)
self.pm.set_constraint('alpha', 9)
self.pm.setTrackParameter('alpha', True)
self.assertEqual(self.pm.isParameterTracked('alpha'), True)
self.assertRaises(DiffcalcException, self.pm.set_constraint, 'alpha', 10)
self.hw.get_position.return_value = 888, 11, 999
self.assertEqual(self.pm.get_constraint('alpha'), 11)
print self.pm.reportAllParameters()
print "**"
print self.ms.reportCurrentMode()
print self.pm.reportParametersUsedInCurrentMode()
self.pm.setTrackParameter('alpha', False)
self.assertEqual(self.pm.isParameterTracked('alpha'), False)
self.assertEqual(self.pm.get_constraint('alpha'), 11)
self.hw.get_position.return_value = 888, 12, 999
self.assertEqual(self.pm.get_constraint('alpha'), 11)
self.pm.set_constraint('alpha', 13)
self.assertEqual(self.pm.get_constraint('alpha'), 13)
