chromatix-team / chromatix

Differentiable wave optics using JAX! Documentation can be found at https://chromatix.readthedocs.io
MIT License
77 stars 13 forks source link

Check shapes sources #138

Closed GJBoth closed 5 months ago

GJBoth commented 5 months ago

Sources now raise an error when giving in wrong amplitude vector, per #112.

This also adds tests for that and reorders them a little bit.