def simple_ring():
ring = [elements.Drift('D1', 1, R1=numpy.eye(6), R2=numpy.eye(6)),
elements.Marker('M1', attr='a_value'), elements.M66('M66'),
elements.Drift('D2', 1, T1=numpy.zeros(6), T2=numpy.zeros(6)),
elements.Drift('D3', 1, R1=numpy.eye(6), R2=numpy.eye(6)),
elements.Drift('D4', 1, T1=numpy.zeros(6), T2=numpy.zeros(6))]
return ring
def test_m66(rin, n):
m = numpy.random.rand(6, 6)
m66 = elements.M66('m66', m)
assert m66.Length == 0
rin[n, 0] = 1e-6
element_pass(m66, rin)
expected = numpy.array([m[0, n], m[1, n], m[2, n], m[3, n], m[4, n],
m[5, n]]).reshape(6, 1) * 1e-6
numpy.testing.assert_equal(rin, expected)