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)