From 46e3d8ce247e57afa9d1d3ad984fd722c9943fe5 Mon Sep 17 00:00:00 2001
From: LukasK13 <spam.lukas.klass@gmail.com>
Date: Wed, 29 Apr 2020 17:19:54 +0200
Subject: [PATCH] Check quantities

---
 esbo_etc/classes/psf/Zemax.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/esbo_etc/classes/psf/Zemax.py b/esbo_etc/classes/psf/Zemax.py
index 3a05187..015d44c 100644
--- a/esbo_etc/classes/psf/Zemax.py
+++ b/esbo_etc/classes/psf/Zemax.py
@@ -13,6 +13,7 @@ class Zemax(IPSF):
     A class for modelling the PSF from a Zemax output file
     """
 
+    @u.quantity_input(wl="length", d_aperture="length")
     def __init__(self, file: str, f_number: float, wl: u.Quantity, d_aperture: u.Quantity):
         """
         Initialize a new PSF from a Zemax file.