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 {
"spectrometer type",
SpectrometerType::IdealSpectrometer.into(),
);
props.set("inverted", false.into());
props
}
impl Default for Spectrometer {
......@@ -98,6 +99,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();
......@@ -214,16 +218,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);
}
}
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