def test_ne(): objs = [ batoid.Paraboloid(1.0), batoid.Paraboloid(2.0), batoid.Plane() ] all_obj_diff(objs)
def test_ne(): objs = [ batoid.Sphere(1.0), batoid.Sphere(2.0), batoid.Plane() ] all_obj_diff(objs)
def test_ne(): objs = [ batoid.Ray((0, 0, 0), (0, 0, 0)), batoid.Ray((0, 0, 0), (0, 0, 0), coordSys=batoid.CoordSys((0, 0, 1))), batoid.Ray((0, 0, 1), (0, 0, 0)), batoid.Ray((0, 1, 0), (0, 0, 0)), batoid.Ray((0, 0, 0), (0, 0, 0), t=1), batoid.Ray((0, 0, 0), (0, 0, 0), wavelength=500e-9), batoid.Ray((0, 0, 0), (0, 0, 0), wavelength=500e-9, flux=1.2), batoid.Ray((0, 0, 0), (0, 0, 0), vignetted=True), batoid.Ray(failed=True), (0, 0, 0), batoid.RayVector([ batoid.Ray((0, 0, 1), (0, 0, 0)), batoid.Ray((0, 0, 0), (0, 0, 0)) ]), batoid.RayVector([ batoid.Ray((0, 0, 1), (0, 0, 0), coordSys=batoid.CoordSys((0, 0, 1))), batoid.Ray((0, 0, 0), (0, 0, 0), coordSys=batoid.CoordSys((0, 0, 1))) ]), batoid.RayVector([ batoid.Ray((0, 0, 0), (0, 0, 0)), batoid.Ray((0, 0, 1), (0, 0, 0)) ]), batoid.RayVector([batoid.Ray((0, 0, 0), (0, 0, 0))]) ] all_obj_diff(objs)
def test_ne(): xs1 = np.linspace(0, 1, 10) xs2 = np.linspace(0, 1, 11) ys1 = np.linspace(0.1, 1, 10) ys2 = np.linspace(0.1, 0.9, 11) def f1(x, y): return x + y def f2(x, y): return x - y zs1 = f1(*np.meshgrid(xs1, ys1)) zs2 = f2(*np.meshgrid(xs1, ys1)) objs = [ batoid.Bicubic(xs1, ys1, zs1), batoid.Bicubic(xs1, ys1, zs2), batoid.Bicubic(xs2, ys1, zs1), batoid.Bicubic(xs1, ys2, zs1), batoid.Bicubic(xs1, ys2, zs1, zs1, zs1, zs1), batoid.Bicubic(xs1, ys2, zs1, zs1, zs1, zs2), batoid.Bicubic(xs1, ys2, zs1, zs1, zs2, zs2), batoid.Bicubic(xs1, ys2, zs1, zs2, zs2, zs2) ] all_obj_diff(objs)
def test_ne(): objs = [ batoid.Plane(), batoid.Tilted(0.0, 0.1), batoid.Tilted(0.1, 0.1), ] all_obj_diff(objs)
def test_ne(): objs = [ batoid.ConstMedium(1.0), batoid.ConstMedium(1.1), batoid.TableMedium([1, 2, 3], [3, 4, 3]), batoid.SellmeierMedium([ 0.6961663, 0.4079426, 0.8974794, 0.0684043**2, 0.1162414**2, 9.896161**2 ]), batoid.SellmeierMedium([ 0.4079426, 0.6961663, 0.8974794, 0.0684043**2, 0.1162414**2, 9.896161**2 ]), batoid.SumitaMedium([ 2.2705778, -0.010059376, 0.010414999, 0.00028872517, -2.2214495e-5, 1.4258559e-6 ]), batoid.SumitaMedium([ -0.010059376, 2.2705778, 0.010414999, 0.00028872517, -2.2214495e-5, 1.4258559e-6 ]), batoid.Air(), batoid.Air(pressure=100) ] all_obj_diff(objs)
def test_ne(): objs = [ batoid.Plane(), batoid.Plane(allowReverse=True), batoid.Paraboloid(2.0), ] all_obj_diff(objs)
def test_ne(): objs = [ batoid.Mirror(batoid.Plane()), batoid.Detector(batoid.Plane()), batoid.Baffle(batoid.Plane()), batoid.RefractiveInterface(batoid.Plane()), batoid.Mirror(batoid.Paraboloid(0.1)), batoid.Detector(batoid.Paraboloid(0.1)), batoid.Baffle(batoid.Paraboloid(0.1)), batoid.RefractiveInterface(batoid.Paraboloid(0.1)), batoid.Mirror(batoid.Plane(), obscuration=batoid.ObscCircle(0.1)), batoid.Mirror(batoid.Plane(), inMedium=batoid.ConstMedium(1.1)), batoid.Mirror(batoid.Plane(), outMedium=batoid.ConstMedium(1.1)), batoid.Mirror(batoid.Plane(), coordSys=batoid.CoordSys([0, 0, 1])), batoid.CompoundOptic( [batoid.Mirror(batoid.Plane()), batoid.Mirror(batoid.Plane())]), batoid.CompoundOptic( [batoid.Mirror(batoid.Plane()), batoid.Baffle(batoid.Plane())]), batoid.CompoundOptic([ batoid.RefractiveInterface(batoid.Plane()), batoid.RefractiveInterface(batoid.Plane()) ]), batoid.Lens([ batoid.RefractiveInterface(batoid.Plane()), batoid.RefractiveInterface(batoid.Plane()) ]), ] all_obj_diff(objs)
def test_ne(): objs = [ batoid.Table([0, 1], [1, 2], 'linear'), batoid.Table([0, 1], [1, 2], 'ceil'), batoid.Table([0, 1], [2, 1], 'linear'), batoid.Table([], [], 'linear') ] all_obj_diff(objs)
def test_ne(): objs = [ batoid.CoordSys(), batoid.CoordTransform(batoid.CoordSys(), batoid.CoordSys()), batoid.CoordTransform(batoid.CoordSys(), batoid.CoordSys(origin=(0, 0, 1))) ] all_obj_diff(objs)
def test_ne(): objs = [ batoid.Table([0, 1], [1, 2], batoid.Table.Interpolant.linear), batoid.Table([0, 1], [1, 2], batoid.Table.Interpolant.ceil), batoid.Table([0, 1], [2, 1], batoid.Table.Interpolant.linear), batoid.Table([], [], batoid.Table.Interpolant.linear) ] all_obj_diff(objs)
def test_ne(): objs = [ batoid.Quadric(10.0, 1.0), batoid.Quadric(11.0, 1.0), batoid.Quadric(10.0, 1.1), batoid.Sphere(10.0) ] all_obj_diff(objs)
def test_ne(): objs = [ batoid.SimpleCoating(0.0, 1.0), batoid.SimpleCoating(0.0, 0.1), batoid.SimpleCoating(0.1, 0.1), batoid.CoordSys() ] all_obj_diff(objs)
def test_ne(): objs = [ batoid.Zernike([0,0,0,0,1]), batoid.Zernike([0,0,0,1]), batoid.Zernike([0,0,0,0,1], R_outer=1.1), batoid.Zernike([0,0,0,0,1], R_inner=0.8), batoid.Zernike([0,0,0,0,1], R_outer=1.1, R_inner=0.8), batoid.Quadric(10.0, 1.0) ] all_obj_diff(objs)
def test_ne(): objs = [ batoid.CoordSys(), batoid.CoordSys([0,0,1]), batoid.CoordSys([0,1,0]), batoid.CoordSys(batoid.RotX(0.1)), batoid.CoordTransform(batoid.CoordSys(), batoid.CoordSys()), batoid.CoordTransform(batoid.CoordSys(), batoid.CoordSys([0,0,1])), batoid.CoordTransform(batoid.CoordSys(), batoid.CoordSys(batoid.RotX(0.1))) ] all_obj_diff(objs)
def test_ne(): objs = [ batoid.Asphere(10.0, 1.0, []), batoid.Asphere(10.0, 1.0, [0]), batoid.Asphere(10.0, 1.0, [0,1]), batoid.Asphere(10.0, 1.0, [1,0]), batoid.Asphere(10.0, 1.1, []), batoid.Asphere(10.1, 1.0, []), batoid.Quadric(10.0, 1.0) ] all_obj_diff(objs)
def test_ne(): objs = [ batoid.ObscCircle(1.0), batoid.ObscCircle(2.0), batoid.ObscCircle(1.0, 0.1, 0.1), batoid.ObscAnnulus(0.0, 1.0), batoid.ObscAnnulus(0.1, 1.0), batoid.ObscAnnulus(0.1, 1.0, 0.1, 0.1), batoid.ObscRectangle(1.0, 2.0), batoid.ObscRectangle(1.0, 2.0, 0.1, 0.1), batoid.ObscRectangle(1.0, 2.0, 0.1, 0.1, 1.0), batoid.ObscRay(1.0, 2.0), batoid.ObscRay(1.0, 2.0, 0.1, 0.1), batoid.ObscNegation(batoid.ObscCircle(1.0)), batoid.ObscPolygon([0,1,1,0],[0,0,1,1]), batoid.ObscUnion([batoid.ObscCircle(1.0)]), batoid.ObscUnion([ batoid.ObscCircle(1.0), batoid.ObscCircle(2.0) ]), batoid.ObscUnion([ batoid.ObscCircle(1.0), batoid.ObscCircle(2.2) ]), batoid.ObscUnion([ batoid.ObscCircle(1.0), batoid.ObscCircle(2.2), batoid.ObscAnnulus(1.0, 2.0) ]), batoid.ObscIntersection([batoid.ObscCircle(1.0)]), batoid.ObscIntersection([ batoid.ObscCircle(1.0), batoid.ObscCircle(2.0) ]), batoid.ObscIntersection([ batoid.ObscCircle(1.0), batoid.ObscCircle(2.2) ]), batoid.ObscIntersection([ batoid.ObscCircle(1.0), batoid.ObscCircle(2.2), batoid.ObscAnnulus(1.0, 2.0) ]), ] all_obj_diff(objs)
def test_ne(): rng = np.random.default_rng(57) N1 = rng.integers(1, 5) N2 = rng.integers(1, 5) N3 = rng.integers(1, 5) arr = np.ones((N1, N2, N3)) pv1 = rng.uniform(-1.0, 1.0, size=3) pv2 = rng.uniform(-1.0, 1.0, size=3) pv3 = rng.uniform(-1.0, 1.0, size=3) lattice1 = batoid.Lattice(arr, np.vstack([pv1, pv2, pv3])) lattice2 = batoid.Lattice(arr[..., 0], np.vstack([pv1, pv2, pv3])[:2, :2]) lattice3 = batoid.Lattice(arr, 2 * np.vstack([pv1, pv2, pv3])) lattice4 = batoid.Lattice(2 * arr, np.vstack([pv1, pv2, pv3])) objs = [batoid.CoordSys(), lattice1, lattice2, lattice3, lattice4] all_obj_diff(objs)
def test_ne(): filename = os.path.join(batoid.datadir, "media", "silica_dispersion.txt") wave, n = np.genfromtxt(filename).T wave *= 1e-6 # microns -> meters table = batoid.Table(wave, n, batoid.Table.Interpolant.linear) table2 = batoid.Table(wave * 1.01, n, batoid.Table.Interpolant.linear) objs = [ batoid.ConstMedium(1.0), batoid.ConstMedium(1.1), batoid.TableMedium(table), batoid.TableMedium(table2), batoid.SellmeierMedium(0.6961663, 0.4079426, 0.8974794, 0.0684043**2, 0.1162414**2, 9.896161**2), batoid.SellmeierMedium(0.4079426, 0.6961663, 0.8974794, 0.0684043**2, 0.1162414**2, 9.896161**2), batoid.Air(), batoid.Air(pressure=100) ] all_obj_diff(objs)
def test_ne(): objs = [ batoid.Ray((0, 0, 0), (0, 0, 0)), batoid.Ray((0, 0, 1), (0, 0, 0)), batoid.Ray((0, 1, 0), (0, 0, 0)), batoid.Ray((0, 0, 0), (0, 0, 0), t=1), batoid.Ray((0, 0, 0), (0, 0, 0), wavelength=500e-9), batoid.Ray((0, 0, 0), (0, 0, 0), wavelength=500e-9, flux=1.2), batoid.Ray((0, 0, 0), (0, 0, 0), vignetted=True), # Should really get a failed Ray to test here... (0, 0, 0), batoid.RayVector(), batoid.RayVector([ batoid.Ray((0, 0, 1), (0, 0, 0)), batoid.Ray((0, 0, 0), (0, 0, 0)) ]), batoid.RayVector([ batoid.Ray((0, 0, 0), (0, 0, 0)), batoid.Ray((0, 0, 1), (0, 0, 0)) ]), batoid.RayVector([batoid.Ray((0, 0, 0), (0, 0, 0))]) ] all_obj_diff(objs)