diff --git a/tests/test_Source/test_accretion_disk_reprocessing.py b/tests/test_Source/test_accretion_disk_reprocessing.py index 04a75d45e..191fc057a 100644 --- a/tests/test_Source/test_accretion_disk_reprocessing.py +++ b/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)