def test_propagating_normal_to_plane(): r1 = Ray(position, wavevector_unit, wavelength, energy) r1.propagate_free_space(10) r2 = Ray(position, wavevector_unit, wavelength, energy) r2.propagate_to_plane([10,10,10], wavevector_unit) assert allclose(r1.position, r2.position)
def test_propagating_angle_to_plane(): r = Ray(position, [3./5,4./5,0], wavelength, energy) r.propagate_to_plane([12,0,0], [1,0,0]) assert allclose(r.position, position + array([12,16,0]))