python source code of test_physics

Project: prysm (GitHub Link)

prysm-master
- Examples
  - Thin Lens Models.ipynb
  - Estimating Effective Pixel Aperture_files
    - MTFMapperParser.py
    - edge_sfr_values_r.txt
  - MTF vs Code V.ipynb
  - LensRentals Blog
    - Lens Performance for High Res Video and Stills
- prysm
  - refractive.py
  - qpoly.py
  - coordinates.py
  - sample_data.py
  - zernike.py
  - util.py
  - convolution.py
  - psf.py
  - thinfilm.py
  - fttools.py
  - _phase.py
  - _richdata.py
  - objects.py
  - jacobi.py
  - geometry.py
  - detector.py
  - pupil.py
  - mtf_utils.py
  - interferogram.py
  - otf.py
  - propagation.py
  - thinlens.py
  - io.py
  - plotting.py
  - __init__.py
  - degredations.py
  - conf.py
  - mathops.py
  - wavelengths.py
- .pydocstyle
- .coveragerc
- sloccounts.csv
- readthedocs.yml
- paper
  - paper.md
  - paper.bib
- setup.py
- LICENSE.md
- sample_files
  - valid_sample_trioptics_mtf_vs_field.mht
- setup.cfg
- .travis.yml
- README.md
- tests
  - test_mtf_utils.py
  - test_degredations.py
  - test_plotting.py
  - test_samplefiles.py
  - test_psf.py
  - .pydocstyle
  - test_detector.py
  - test_thinfilm.py
  - test_coordinates.py
  - test_io.py
  - test_qpoly.py
  - test_physics.py
  - test_jacobi.py
  - test_convolution.py
  - test_otf.py
  - test_e2e.py
  - test_util.py
  - test_richdata.py
  - test_objects.py
  - test_propagation.py
  - test_config.py
  - test_pupil.py
  - test_refactive.py
  - __init__.py
  - test_geometry.py
  - test_zernike.py
  - test_fttools.py
  - test_interferogram.py
  - test_mathops.py
  - test_thinlens.py
- .gitignore
- docs
  - Makefile
  - source
    - examples
      - System Model.ipynb
      - Defocus and Contrast Inversion.ipynb
      - Diffraction Limited PSF and MTF.ipynb
      - Numerically Calculated Aberration Transfer Function.ipynb
      - Onion Ring Bokeh.ipynb
      - Split Transform Method.ipynb
      - Analysis of Interferometric Wavefront Data.ipynb
      - index.rst
    - api
      - wavelengths.rst
      - otf.rst
      - coordinates.rst
      - sample_data.rst
      - mtf_utils.rst
      - mathops.rst
      - io.rst
      - conf.rst
      - fttools.rst
      - geometry.rst
      - propagation.rst
      - util.rst
      - objects.rst
      - base_classes.rst
      - qpoly.rst
      - plotting.rst
      - detector.rst
      - refractive.rst
      - degredations.rst
      - zernike.rst
      - pupil.rst
      - index.rst
      - thinfilm.rst
      - convolution.rst
      - psf.rst
      - interferogram.rst
      - jacobi.rst
      - thinlens.rst
    - releases
      - v0.17.rst
      - v0.15.rst
      - v0.19.rst
      - v0.16.1.rst
      - v0.14.rst
      - v0.13.rst
      - v0.16.rst
      - v0.17.2.rst
      - index.rst
      - v0.18.rst
    - index.rst
    - conf.py
    - _static
    - contributing.rst
    - user_guide
      - Pupils.ipynb
      - MTFs.ipynb
      - Zernikes.ipynb
      - PSFs.ipynb
      - plotting.ipynb
      - Convolvables.ipynb
      - GPU and high speed computing.ipynb
      - index.rst
      - Interferograms.ipynb
      - slicing.ipynb
      - units-and-labels.ipynb
      - Conventions.ipynb
  - make.bat
  - requirements.txt
- MANIFEST.in

"""Unit tests for the physics of prysm."""
from itertools import product

import numpy as np

from astropy import units as u

import pytest

from prysm.wavelengths import mkwvl
from prysm import Pupil, PSF, MTF, FringeZernike
from prysm.psf import airydisk
from prysm.otf import diffraction_limited_mtf

PRECISION = 1e-3  # ~0.1%

TEST_PARAMETERS = [
    (10.0, 1.000, 0.5),  # f/10, visible light
    (10.0, 1.000, 1.0),  # f/10, SWIR light
    (3.00, 1.125, 3.0)]  # f/2.66666, MWIR light


@pytest.mark.parametrize('efl, epd, wvl', TEST_PARAMETERS)
def test_diffprop_matches_airydisk(efl, epd, wvl):
    fno = efl / epd

    p = Pupil(dia=epd, xy_unit=u.mm, z_unit=u.nm, wavelength=mkwvl(wvl, u.um))
    psf = PSF.from_pupil(p, efl, Q=3)  # use Q=3 not Q=4 for improved accuracy
    s = psf.slices()
    u_, sx = s.x
    u_, sy = s.y
    analytic = airydisk(u_, fno, wvl)
    assert np.allclose(sx, analytic, atol=PRECISION)
    assert np.allclose(sy, analytic, atol=PRECISION)


@pytest.mark.parametrize('efl, epd, wvl', TEST_PARAMETERS)
def test_diffprop_matches_analyticmtf(efl, epd, wvl):
    fno = efl / epd
    p = Pupil(dia=epd, xy_unit=u.mm, z_unit=u.nm, wavelength=mkwvl(wvl, u.um))
    psf = PSF.from_pupil(p, efl)
    mtf = MTF.from_psf(psf)
    s = mtf.slices()
    u_, x = s.x
    u__, y = s.y

    analytic_1 = diffraction_limited_mtf(fno, wvl, frequencies=u_)
    analytic_2 = diffraction_limited_mtf(fno, wvl, frequencies=u__)
    assert np.allclose(analytic_1, x, atol=PRECISION)
    assert np.allclose(analytic_2, y, atol=PRECISION)


def test_array_orientation_consistency_tilt():
    """The pupil array should be shaped as arr[y,x], as should the psf and MTF.

        A linear phase error in the pupil along y should cause a motion of the
        PSF in y.  Specifically, for a positive-signed phase, that should cause
        a shift in the +y direction.
    """
    samples = 128
    p = FringeZernike(Z2=1000, samples=samples)
    ps = PSF.from_pupil(p, 1)
    idx_y, idx_x = np.unravel_index(ps.data.argmax(), ps.data.shape)  # row-major y, x
    assert idx_x == ps.center_x
    assert idx_y > ps.center_y


FNOS = [1, 1.4, 2, 2.8, 4, 5.6, 8]
WVLS = [.5, .55, 1, 10]


@pytest.mark.parametrize('fno, wvl', product(FNOS, WVLS))
def test_airydisk_has_unit_peak(fno, wvl):
    assert airydisk(0, fno=fno, wavelength=wvl) == 1