예제 #1
0
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)
예제 #2
0
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]))