Skip to content
Snippets Groups Projects
Commit 9db9a05e authored by Udo Eisenbarth's avatar Udo Eisenbarth :speech_balloon:
Browse files

Add unit tests for Spectrometer::analyze

parent 3e95df99
No related branches found
No related tags found
1 merge request!29Resolve "Spectrometer: Add analysis unit test"
...@@ -53,6 +53,7 @@ fn create_default_props() -> Properties { ...@@ -53,6 +53,7 @@ fn create_default_props() -> Properties {
"spectrometer type", "spectrometer type",
SpectrometerType::IdealSpectrometer.into(), SpectrometerType::IdealSpectrometer.into(),
); );
props.set("inverted", false.into());
props props
} }
impl Default for Spectrometer { impl Default for Spectrometer {
...@@ -98,6 +99,9 @@ impl Optical for Spectrometer { ...@@ -98,6 +99,9 @@ impl Optical for Spectrometer {
fn node_type(&self) -> &str { fn node_type(&self) -> &str {
"spectrometer" "spectrometer"
} }
fn inverted(&self) -> bool {
self.properties().get_bool("inverted").unwrap().unwrap()
}
fn ports(&self) -> OpticPorts { fn ports(&self) -> OpticPorts {
let mut ports = OpticPorts::new(); let mut ports = OpticPorts::new();
ports.add_input("in1").unwrap(); ports.add_input("in1").unwrap();
...@@ -214,16 +218,61 @@ mod test { ...@@ -214,16 +218,61 @@ mod test {
assert_eq!(meter.ports().outputs(), vec!["out1"]); assert_eq!(meter.ports().outputs(), vec!["out1"]);
} }
#[test] #[test]
fn analyze() { fn inverted() {
let mut meter = Spectrometer::default(); 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(); let mut input = LightResult::default();
input.insert( let input_light = LightData::Energy(DataEnergy {
"in1".into(), spectrum: create_he_ne_spectrum(1.0),
Some(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 result = meter.analyze(input, &AnalyzerType::Energy); let output = output.unwrap();
assert!(result.is_ok()); 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);
} }
} }
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment