Add opticalsurface component and unit test

architect/systems/optical/optical_surface.py

@@ -0,0 +1,30 @@
+"""An optical surface component."""
+
+# external
+import astropy.units as unit
+import numpy as np
+
+# project
+from architect.systems import Component
+
+
+class OpticalSurface(Component):
+    """An optical surface is a component that can refract light."""
+
+    def __init__(self, medium_index=None):
+        super().__init__()
+        self.medium_index = medium_index
+
+    def angle_of_refraction(self, incident_angle: unit.deg, incident_index) -> unit.deg:
+        """Get the angle of refraction for a given incident angle.
+
+        Ref: https://www.notion.so/utat-ss/Angle-of-Refraction-a4c5f3235ad941398137dd3560eec717
+
+        """
+        assert self.medium_index is not None, "medium_index must be set."
+
+        angle_of_refraction = np.arcsin(
+            incident_index * np.sin(incident_angle) / self.medium_index
+        )
+
+        return angle_of_refraction.to(unit.deg)

tests/test_systems/test_optical/test_OpticalSurface.py

@@ -0,0 +1,35 @@
+"""Tests for OpticalSurface."""
+
+# stdlib
+import logging
+
+# external
+import astropy.units as unit
+
+# project
+from architect.systems.optical.optical_surface import OpticalSurface
+
+LOG = logging.getLogger(__name__)
+
+
+def test_init():
+    """Test init method."""
+
+    component = OpticalSurface()
+    LOG.info(component)
+
+
+def test_angle_of_refraction():
+    """Test angle_of_refraction method."""
+
+    component = OpticalSurface(medium_index=1.5)
+
+    incident_angle = 45 * unit.deg
+    incident_index = 1.0
+    angle_of_refraction = component.angle_of_refraction(
+        incident_angle=incident_angle, incident_index=incident_index
+    )
+
+    LOG.info(angle_of_refraction)
+
+    assert angle_of_refraction.unit == unit.deg