Add missing sensor component unit tests

.pre-commit-config.yaml

@@ -1,26 +1,26 @@
 repos:
 - repo: https://github.com/pycqa/isort
- rev: 5.10.1
+ rev: 5.12.0
  hooks:
  - id: isort
 
 - repo: https://github.com/psf/black
- rev: 22.3.0
+ rev: 23.7.0
  hooks:
  - id: black
 
 - repo: https://github.com/PyCQA/autoflake
- rev: v1.7.7
+ rev: v2.2.0
  hooks:
  - id: autoflake
 
 - repo: https://github.com/asottile/pyupgrade
- rev: v2.32.0
+ rev: v3.9.0
  hooks:
  - id: pyupgrade
 
 - repo: https://github.com/myint/docformatter
- rev: v1.5.0
+ rev: v1.7.5
  hooks:
  - id: docformatter
  args: [--blank, --in-place, --recursive, --wrap-descriptions=88, --wrap-summaries=80] # does not yet support toml config
@@ -33,7 +33,7 @@ repos:
 
 
 - repo: https://github.com/nbQA-dev/nbQA
- rev: 1.3.1
+ rev: 1.7.0
  hooks:
  - id: nbqa-isort
  - id: nbqa-black

tests/test_systems/test_optical/test_sensors/test_Sensor.py

@@ -66,6 +66,18 @@ def test_get_shape():
  assert result[1].decompose().unit == unit.m
 
 
+def test_get_noise_read():
+ """Test get_noise_read method."""
+
+ noise_read = 500 * unit.electron
+ sensor = Sensor(noise_read=noise_read)
+
+ result = sensor.get_noise_read()
+ LOG.info(result)
+
+ assert result == noise_read
+
+
 def test_get_area():
  """Test get_area method."""
 
@@ -102,6 +114,20 @@ def test_get_mean_dark_signal():
  assert result.decompose().unit == (unit.electron / unit.pix)
 
 
+def test_get_dark_shot_noise():
+ """Test get_dark_shot_noise method."""
+
+ ke = 1e3 * unit.electron
+ sensor = Sensor(
+ integration_time=166.7 * unit.ms, i_dark=140 * (ke / unit.pix / unit.s)
+ )
+
+ result = sensor.get_dark_shot_noise()
+ LOG.info(result)
+
+ assert result.decompose().unit == (unit.electron / unit.pix) ** 0.5
+
+
 def test_get_quantization_noise():
  """Test get_quantization_noise method."""
 
@@ -156,3 +182,15 @@ def test_get_efficiency():
  LOG.info(result)
 
  assert result.unit == unit.pct * unit.electron
+
+
+def test_get_waveband():
+ """Test get_waveband method."""
+
+ waveband = 800 * unit.nm
+ sensor = Sensor(waveband=waveband)
+
+ result = sensor.get_waveband()
+ LOG.info(result)
+
+ assert result == waveband