Merge branch '82-spectrometer-add-analysis-unit-test' into 'main'

Resolve "Spectrometer: Add analysis unit test"

Closes #82

See merge request !29

src/nodes/spectrometer.rs

@@ -53,9 +53,11 @@ fn create_default_props() -> Properties {
         "spectrometer type",
         SpectrometerType::IdealSpectrometer.into(),
     );
+    props.set("inverted", false.into());
     props
 }
 impl Default for Spectrometer {
+    /// create an ideal spectrometer.
     fn default() -> Self {
         Self {
             light_data: None,
@@ -65,9 +67,10 @@ impl Default for Spectrometer {
 }
 impl Spectrometer {
     /// Creates a new [`Spectrometer`] of the given [`SpectrometerType`].
-    pub fn new(spectrometer_type: SpectrometerType) -> Self {
+    pub fn new(name: &str, spectrometer_type: SpectrometerType) -> Self {
         let mut props = create_default_props();
         props.set("spectrometer type", spectrometer_type.into());
+        props.set("name", name.into());
         Spectrometer {
             props,
             ..Default::default()
@@ -98,6 +101,9 @@ impl Optical for Spectrometer {
     fn node_type(&self) -> &str {
         "spectrometer"
     }
+    fn inverted(&self) -> bool {
+        self.properties().get_bool("inverted").unwrap().unwrap()
+    }
     fn ports(&self) -> OpticPorts {
         let mut ports = OpticPorts::new();
         ports.add_input("in1").unwrap();
@@ -197,13 +203,14 @@ mod test {
     }
     #[test]
     fn new() {
-        let meter = Spectrometer::new(SpectrometerType::HR2000);
+        let meter = Spectrometer::new("test", SpectrometerType::HR2000);
+        assert_eq!(meter.name(), "test");
         assert!(meter.light_data.is_none());
         assert_eq!(meter.spectrometer_type(), SpectrometerType::HR2000);
     }
     #[test]
     fn set_meter_type() {
-        let mut meter = Spectrometer::new(SpectrometerType::IdealSpectrometer);
+        let mut meter = Spectrometer::new("test", SpectrometerType::IdealSpectrometer);
         meter.set_spectrometer_type(SpectrometerType::HR2000);
         assert_eq!(meter.spectrometer_type(), SpectrometerType::HR2000);
     }
@@ -214,16 +221,61 @@ mod test {
         assert_eq!(meter.ports().outputs(), vec!["out1"]);
     }
     #[test]
-    fn analyze() {
-        let mut meter = Spectrometer::default();
+    fn inverted() {
+        let mut node = Spectrometer::default();
+        node.set_property("inverted", true.into()).unwrap();
+        assert_eq!(node.inverted(), true)
+    }
+    #[test]
+    fn analyze_ok() {
+        let mut node = Spectrometer::default();
         let mut input = LightResult::default();
-        input.insert(
-            "in1".into(),
-            Some(LightData::Energy(DataEnergy {
-                spectrum: create_he_ne_spectrum(1.0),
-            })),
-        );
-        let result = meter.analyze(input, &AnalyzerType::Energy);
-        assert!(result.is_ok());
+        let input_light = LightData::Energy(DataEnergy {
+            spectrum: create_he_ne_spectrum(1.0),
+        });
+        input.insert("in1".into(), Some(input_light.clone()));
+        let output = node.analyze(input, &AnalyzerType::Energy);
+        assert!(output.is_ok());
+        let output = output.unwrap();
+        assert!(output.contains_key("out1".into()));
+        assert_eq!(output.len(), 1);
+        let output = output.get("out1".into()).unwrap();
+        assert!(output.is_some());
+        let output = output.clone().unwrap();
+        assert_eq!(output, input_light);
+    }
+    #[test]
+    fn analyze_wrong() {
+        let mut node =Spectrometer::default();
+        let mut input = LightResult::default();
+        let input_light = LightData::Energy(DataEnergy {
+            spectrum: create_he_ne_spectrum(1.0),
+        });
+        input.insert("wrong".into(), Some(input_light.clone()));
+        let output = node.analyze(input, &AnalyzerType::Energy);
+        assert!(output.is_ok());
+        let output = output.unwrap();
+        let output = output.get("out1".into()).unwrap();
+        assert!(output.is_none());
+    }
+    #[test]
+    fn analyze_inverse() {
+        let mut node = Spectrometer::default();
+        node.set_property("inverted", true.into()).unwrap();
+        let mut input = LightResult::default();
+        let input_light = LightData::Energy(DataEnergy {
+            spectrum: create_he_ne_spectrum(1.0),
+        });
+        input.insert("out1".into(), Some(input_light.clone()));
+
+        let output = node.analyze(input, &AnalyzerType::Energy);
+        assert!(output.is_ok());
+        let output = output.unwrap();
+        assert!(output.contains_key("in1".into()));
+        assert_eq!(output.len(), 1);
+        let output = output.get("in1".into()).unwrap();
+        assert!(output.is_some());
+        let output = output.clone().unwrap();
+        assert_eq!(output, input_light);
     }
 }