100% codecov

tests/test_Source/test_accretion_disk_reprocessing.py

@@ -23,6 +23,14 @@ def test_initialization(self):
         }
         reprocessor = AccretionDiskReprocessing("lamppost", **kwargs_agn_model)
         assert reprocessor.reprocessing_model == "lamppost"
+        assert reprocessor.redshift == 0
+
+        kwargs_agn_model["redshift"] = 3.1
+        redshifted_reprocessor = AccretionDiskReprocessing(
+            "lamppost", **kwargs_agn_model
+        )
+        assert redshifted_reprocessor.redshift == 3.1
+
         with pytest.raises(ValueError):
             AccretionDiskReprocessing("other", **kwargs_agn_model)
 
@@ -43,6 +51,9 @@ def test_initialization_invalid_model(self):
             "Given model is not supported. Currently supported model is lamppost."
         ) in str(excinfo.value)
 
+        kwargs_agn_model["redshift"] = 3.1
+        AccretionDiskReprocessing("lamppost", **kwargs_agn_model)
+
     def test_default_initialization_lamppost_model(self):
         kwargs_agn_model = {"r_out": 1000}
         reprocessor = AccretionDiskReprocessing("lamppost", **kwargs_agn_model)