remove wavelength

Author: roytsmart

optika/materials/_snells_law.py

@@ -39,20 +39,17 @@ def snells_law_scalar(
 
 
 def snells_law(
-    wavelength: u.Quantity | na.AbstractScalar,
     direction: na.AbstractCartesian3dVectorArray,
     index_refraction: float | na.AbstractScalar,
     index_refraction_new: float | na.AbstractScalar,
-    normal: None | na.AbstractCartesian3dVectorArray,
+    normal: None | na.AbstractCartesian3dVectorArray = None,
     is_mirror: bool | na.AbstractScalar = False,
 ) -> na.Cartesian3dVectorArray:
     r"""
     A `vector form of Snell's law <https://en.wikipedia.org/wiki/Snell%27s_law#Vector_form>`_.
 
     Parameters
     ----------
-    wavelength
-        The wavelength of the incoming light
     direction
         The propagation direction of the incoming light
     index_refraction
@@ -91,7 +88,6 @@ def snells_law(
         # Define the keyword arguments that are common
         # to both the reflected and transmitted ray
         kwargs = dict(
-            wavelength=350 * u.nm,
             direction=direction,
             index_refraction=1,
             normal=na.Cartesian3dVectorArray(0, 0, 1),

optika/materials/_tests/test_snells_law.py

@@ -46,13 +46,6 @@ def test_snells_law_scalar(
     assert np.allclose(result, result_expected)
 
 
-@pytest.mark.parametrize(
-    argnames="wavelength",
-    argvalues=[
-        350 * u.nm,
-        na.linspace(300 * u.nm, 400 * u.nm, axis="wavelength", num=3),
-    ],
-)
 @pytest.mark.parametrize(
     argnames="direction",
     argvalues=[
@@ -87,15 +80,13 @@ def test_snells_law_scalar(
     ],
 )
 def test_snells_law(
-    wavelength: u.Quantity | na.AbstractScalar,
     direction: na.AbstractCartesian3dVectorArray,
     index_refraction: float | na.AbstractScalar,
     index_refraction_new: float | na.AbstractScalar,
     normal: None | na.AbstractCartesian3dVectorArray,
     is_mirror: bool | na.AbstractScalar,
 ):
     result = optika.materials.snells_law(
-        wavelength=wavelength,
         direction=direction,
         index_refraction=index_refraction,
         index_refraction_new=index_refraction_new,

optika/surfaces.py

@@ -163,7 +163,6 @@ def propagate_rays(
         wavelength_2 = wavelength_1 / r
 
         b = optika.materials.snells_law(
-            wavelength=wavelength_1,
             direction=a,
             index_refraction=n1,
             index_refraction_new=n2,