def test_createScan(self): a, b, c, mied = createScannables() scanobject = TrajScan()._createScan([a, 1, 10, 1, b, 2, c]) self.MockTrajectoryScanLine.assert_called_with([a, 1, 10, 1, b, 2, c]) self.assertEqual(scanobject, self.mockTrajectoryScanLine)
def test_createScan(self): a, b, c, mied = createScannables() scanobject = TrajScan()._createScan([a, 1, 10, 1, b, 2, c]) self.MockTrajectoryScanLine.assert_called_with([a, 1, 10, 1, b, 2, c]) self.assertEqual(scanobject, self.mockTrajectoryScanLine)
def test_createNestedOnceScanWithTuple(self): a, b, c, mie = createScannables() scanobject = TrajScan()._createScan([a, 1, 10, 1, b, (1,2,3), c, 1, mie]) self.MockTrajectoryScanLine.assert_called_with([b, (1,2,3), c, 1, mie]) self.MockConcurrentScan.assert_called_with([a, 1, 10, 1, self.mockTrajectoryScanLine]) self.assertEqual(scanobject, self.mockConcurrentScan)
def testCheckArgStructForLoops(self): scan = SpecScan(None) a, b, c, mie = createScannables() scan.checkArgStructForLoops([[a], [b, 1], [c], [mie, (1, 2)]]) self.assertRaises(Exception, scan.checkArgStructForLoops, [[a], [b, 1], [c], [mie, (1, 2), (1, 2)]]) self.assertRaises(Exception, scan.checkArgStructForLoops, [[a], [b, 1], [c, 1, 2, 3], [mie, (1, 2), (1, 2)]])
def test_createNestedOnceScanWithTuple(self): a, b, c, mie = createScannables() scanobject = TrajScan()._createScan( [a, 1, 10, 1, b, (1, 2, 3), c, 1, mie]) self.MockTrajectoryScanLine.assert_called_with( [b, (1, 2, 3), c, 1, mie]) self.MockConcurrentScan.assert_called_with( [a, 1, 10, 1, self.mockTrajectoryScanLine]) self.assertEqual(scanobject, self.mockConcurrentScan)
def testConvertArgsToConcurrentScanArgs(self): convert = self.scan.convertArgStruct a, b, c, _ = createScannables() # ascan motor start finish intervals [scn [pos]]... self.assertEqual(convert([[a, 1, 10, 9]]), [[a, 1, 10, 1]]) self.assertEqual(convert([[a, 10, 1, 9]]), [[a, 10, 1, -1]]) self.assertEqual(convert([[a, 1, 10, 9], [b, 1], [c]]), [[a, 1, 10, 1], [b, 1], [c]]) self.assertRaises(Exception, convert, [[a, 1, 10, 9], [b, 1, 2, 3], [c]]) self.assertRaises(Exception, convert, [[a, 1, 10, 9], [b, 1, 2], [c]])
def testConvertArgsToConcurrentScanArgs(self): convert = self.scan.convertArgStruct a, b, c, _ = createScannables() # cscan motor halfwidths step [motor halfwidth step] ... self.assertEqual(convert([[a, 10, 1]]), [[a, -10, 10, 1]]) self.assertEqual(convert([[a, 10, 1], [b, 5, 2]]), [[a, -10, 10, 1], [b, -5, 5, 2]]) self.assertEqual(convert([[a, 10, 1], [b, 1]]), [[a, -10, 10, 1], [b, 1]]) self.assertEqual(convert([[a, 10, 1], [b, 1], [c]]), [[a, -10, 10, 1], [b, 1], [c]])
def testConvertArgsToConcurrentScanArgsWithMultipleInputFields(self): convert = self.scan.convertArgStruct a, b, c, _ = createScannables() # cscan motor halfwidths step [motor halfwidth step] ... self.assertEqual(convert([[a, [10, 100], [1, 10]]]), [[a, [-10, -100], [10, 100], [1, 10]]]) self.assertEqual(convert([[a, [10, 100], [1, 10]], [b, 5, 2]]), [[a, [-10, -100], [10, 100], [1, 10]], [b, -5, 5, 2]]) self.assertEqual(convert([[a, [10, 100], [1, 10]], [b, 1]]), [[a, [-10, -100], [10, 100], [1, 10]], [b, 1]]) self.assertEqual(convert([[a, [10, 100], [1, 10]], [b, 1], [c]]), [[a, [-10, -100], [10, 100], [1, 10]], [b, 1], [c]])
def testConvertArgsToConcurrentScanArgs(self): convert = self.scan.convertArgStruct a, b, c, _ = createScannables() # cscan motor halfwidths step [motor halfwidth step] ... self.assertEqual(convert([[a, .2, 10]]), [[a, -.9, .9, .2]]) self.assertEqual(convert([[a, .2, 9]]), [[a, -.8, .8, .2]]) self.assertEqual(convert([[a, .2, 9], [b, 1, 2]]), [[a, -.8, .8, .2], [ b, 1, 2, ]])
def testConvertArgsToConcurrentScanArgsWithMultipleInputFields(self): convert = self.scan.convertArgStruct a, b, c, _ = createScannables() # cscan motor halfwidths step [motor halfwidth step] ... self.assertEqual(convert([[a, [.2, 20], 10]]), [[a, [-.9, -90], [.9, 90], [.2, 20]]]) self.assertEqual(convert([[a, [.2, 20], 9]]), [[a, [-.8, -80], [.8, 80], [.2, 20]]]) self.assertEqual(convert([[a, [.2, 20], 9], [b, 1, 2]]), [[a, [-.8, -80], [.8, 80], [.2, 20]], [ b, 1, 2, ]])
def testConvertArgsToConcurrentScanArgs(self): convert = self.scan.convertArgStruct a, b, c, _ = createScannables() #mesh m1 s1 f1 intervals1 m2 s2 f2 intervals2 time ... self.assertEqual(convert([[a, 1, 10, 18], [b, 11, 20, 9]]), [[a, 1, 10, .5], [b, 11, 20, 1]]) self.assertEqual(convert([[a, 1, 10, 18], [b, 20, 11, 9]]), [[a, 1, 10, .5], [b, 20, 11, -1]]) self.assertEqual( convert([[a, 1, 10, 18], [b, 11, 20, 9], [b, 1], [c]]), [[a, 1, 10, .5], [b, 11, 20, 1], [b, 1], [c]]) self.assertRaises(Exception, convert, [[a, 1, 10, 18], [b, 11, 20, 9], [b, 1, 2], [c]]) self.assertRaises(Exception, convert, [[a, 1, 10, 18], [b, 11, 20, 9], [b, 1, 2, 3], [c]])
def testConvertArgsToConcurrentScanArgs(self): convert = self.scan.convertArgStruct a, b, c, _ = createScannables() # a3scan m1 s1 f1 m2 s2 f2 m3 s3 f3 intervals [scn [pos]] ... self.assertEqual(convert([[a, 1, 10], [b, 11, 20], [c, 21, 30, 9]]), [[a, 1, 10, 1], [b, 11, 20, 1], [c, 21, 30, 1]]) self.assertEqual(convert([[a, 10, 1], [b, 11, 20], [c, 30, 21, 9]]), [[a, 10, 1, -1], [b, 11, 20, 1], [c, 30, 21, -1]]) self.assertEqual( convert([[a, 10, 1], [b, 11, 20], [c, 30, 21, 9], [b, 1], [c]]), [[a, 10, 1, -1], [b, 11, 20, 1], [c, 30, 21, -1], [b, 1], [c]]) self.assertRaises( Exception, convert, [[a, 10, 1], [b, 11, 20], [c, 30, 21, 9], [b, 1, 2], [c]]) self.assertRaises( Exception, convert, [[a, 10, 1], [b, 11, 20], [c, 30, 21, 9], [b, 1, 2, 3], [c]])
def testConvertArgsToConcurrentScanArgs(self): # Well tested in testConcurrentScanWrapper already! a, b, c, _ = createScannables() argStruct = [[a, 1, 2, 1], [b, 3, 5, 1], [c]] self.assertEqual(self.scan.convertArgStruct(argStruct), argStruct)
def setUp(self): self.a, self.b, self.c, self.mie = createScannables() self.scan = SecondaryConcurrentScan([self.a, 1, 3, 1, self.b, 4, self.c])
def test_createScanWithTuple(self): a, b, c, _ = createScannables() scanobject = TrajScan()._createScan([a, (1, 2, 3), b, 2, c]) self.MockTrajectoryScanLine.assert_called_with([a, (1, 2, 3), b, 2, c]) self.assertEqual(scanobject, self.mockTrajectoryScanLine)
def test_createScan(self): a, b, c, _ = createScannables() scanobject = Scan()._createScan([a, 1, 10, 1, b, 2, c]) self.assertEqual(scanobject, self.mockConcurrentScan) self.MockConcurrentScan.assert_called_with([a, 1, 10, 1, b, 2, c])
def setUp(self): self.a, self.b, self.c, _ = createScannables() self.mie = MultiInputExtraFieldsDummyPD('mie', ['i1'], ['e1','e2']) self.col = SelectableCollectionOfScannables('c', [self.a, self.b, self.mie, self.c])
def test_createScanWithTuple(self): a, b, c, _ = createScannables() scanobject = TrajScan()._createScan([a, (1, 2, 3), b, 2, c]) self.MockTrajectoryScanLine.assert_called_with([a, (1, 2, 3), b, 2, c]) self.assertEqual(scanobject, self.mockTrajectoryScanLine)
def test_createScan(self): a, b, c, _ = createScannables() scanobject = Scan()._createScan([a, 1, 10, 1, b, 2, c]) self.assertEqual(scanobject, self.mockConcurrentScan) self.MockConcurrentScan.assert_called_with([a, 1, 10, 1, b, 2, c])
def setUp(self): self.a, self.b, self.c, _ = createScannables() self.mie = MultiInputExtraFieldsDummyPD('mie', ['i1'], ['e1', 'e2']) self.col = SelectableCollectionOfScannables( 'c', [self.a, self.b, self.mie, self.c])