From eeb10d2897f9c0c9c98f7e803a1d534b183a4a75 Mon Sep 17 00:00:00 2001
From: narayan <nkhadka@ksu.edu>
Date: Thu, 7 Nov 2024 09:25:47 -0500
Subject: [PATCH] added test functions for FalsePositive class.

---
 .../test_false_positive.py                    | 158 ++++++++++++++++++
 1 file changed, 158 insertions(+)

diff --git a/tests/test_FalsePositives/test_false_positive.py b/tests/test_FalsePositives/test_false_positive.py
index e69de29bb..e4ae372c2 100644
--- a/tests/test_FalsePositives/test_false_positive.py
+++ b/tests/test_FalsePositives/test_false_positive.py
@@ -0,0 +1,158 @@
+import pytest
+import numpy as np
+from astropy.cosmology import FlatLambdaCDM
+from slsim.ParamDistributions.los_config import LOSConfig
+from slsim.Sources.source import Source
+from slsim.Deflectors.deflector import Deflector
+import slsim.Sources as sources
+import slsim.Deflectors as deflectors
+import slsim.Pipelines as pipelines
+from slsim.FalsePositives.false_positive import FalsePositive
+from astropy.units import Quantity
+
+sky_area = Quantity(value=0.01, unit="deg2")
+galaxy_simulation_pipeline = pipelines.SkyPyPipeline(
+    skypy_config=None,
+    sky_area=sky_area,
+    filters=None,
+)
+kwargs_deflector_cut = {"band": "g", "band_max": 28, "z_min": 0.01, "z_max": 2.5}
+kwargs_source_cut = {"band": "g", "band_max": 28, "z_min": 0.1, "z_max": 5.0}
+
+
+def test_false_positive():
+    # Mock objects for source_class and deflector_class
+
+    # Initialize a cosmology instance
+    cosmo = FlatLambdaCDM(H0=70, Om0=0.3)
+    lens_galaxies = deflectors.EllipticalLensGalaxies(
+        galaxy_list=galaxy_simulation_pipeline.red_galaxies,
+        kwargs_cut=kwargs_deflector_cut,
+        kwargs_mass2light=0.1,
+        cosmo=cosmo,
+        sky_area=sky_area,
+    )
+    source_galaxies = sources.Galaxies(
+        galaxy_list=galaxy_simulation_pipeline.blue_galaxies,
+        kwargs_cut=kwargs_source_cut,
+        cosmo=cosmo,
+        sky_area=sky_area,
+        catalog_type="skypy",
+    )
+    single_deflector = lens_galaxies.draw_deflector()
+    single_source1 = source_galaxies.draw_source()
+    single_source2 = source_galaxies.draw_source()
+    lens = Deflector(deflector_type="EPL", deflector_dict=single_deflector)
+    source = Source(
+        source_dict=single_source1,
+        cosmo=cosmo,
+        source_type="extended",
+        light_profile="single_sersic",
+    )
+    source2 = Source(
+        source_dict=single_source1,
+        cosmo=cosmo,
+        source_type="extended",
+        light_profile="double_sersic",
+    )
+    source_list = [
+        Source(
+            source_dict=single_source1,
+            cosmo=cosmo,
+            source_type="extended",
+            light_profile="single_sersic",
+        ),
+        Source(
+            source_dict=single_source2,
+            cosmo=cosmo,
+            source_type="extended",
+            light_profile="single_sersic",
+        ),
+    ]
+    # LOS configuration
+    los_config = LOSConfig()
+
+    # Create an instance of FalsePositive
+    false_positive_instance_1 = FalsePositive(
+        source_class=source,
+        deflector_class=lens,
+        cosmo=cosmo,
+        test_area=4 * np.pi,
+    )
+    false_positive_instance_2 = FalsePositive(
+        source_class=source_list,
+        deflector_class=lens,
+        cosmo=cosmo,
+        test_area=4 * np.pi,
+        los_config=los_config,
+    )
+    false_positive_instance_3 = FalsePositive(
+        source_class=source2,
+        deflector_class=lens,
+        cosmo=cosmo,
+        test_area=4 * np.pi,
+        los_config=los_config,
+    )
+    required_keys = {
+        "magnitude",
+        "R_sersic",
+        "n_sersic",
+        "e1",
+        "e2",
+        "center_x",
+        "center_y",
+    }
+    assert false_positive_instance_1.source_number == 1
+    assert false_positive_instance_2.source_number == 2
+    assert (
+        false_positive_instance_1.lenstronomy_kwargs("i")[0]["lens_light_model_list"][0]
+        == "SERSIC_ELLIPSE"
+    )
+    assert (
+        len(
+            false_positive_instance_2.lenstronomy_kwargs("i")[0][
+                "lens_light_model_list"
+            ]
+        )
+        == 3
+    )
+    assert (
+        len(false_positive_instance_2.lenstronomy_kwargs("i")[1]["kwargs_lens_light"])
+        == 3
+    )
+    assert len(false_positive_instance_2.deflector_position) == 2
+    assert false_positive_instance_2.deflector_redshift == single_deflector["z"]
+    assert false_positive_instance_1.source_redshift == single_source1["z"]
+    assert np.all(false_positive_instance_2.source_redshift) == np.all(
+        np.array([single_source1["z"], single_source2["z"]])
+    )
+    assert false_positive_instance_1.external_convergence < 0.1
+    assert false_positive_instance_1.external_shear < 0.2
+    assert false_positive_instance_1.einstein_radius < 2.5
+    assert (
+        false_positive_instance_1.deflector_magnitude(band="i")
+        == single_deflector["mag_i"]
+    )
+    assert (
+        false_positive_instance_1.extended_source_magnitude(band="i")
+        == single_source1["mag_i"]
+    )
+    assert len(false_positive_instance_1.deflector_ellipticity()) == 4
+    assert (
+        false_positive_instance_1.deflector_stellar_mass()
+        == single_deflector["stellar_mass"]
+    )
+    assert (
+        set(
+            false_positive_instance_1.deflector_light_model_lenstronomy(band="i")[1][
+                0
+            ].keys()
+        )
+        == required_keys
+    )
+    with pytest.raises(ValueError):
+        false_positive_instance_3.source_light_model_lenstronomy(band="i")
+
+
+if __name__ == "__main__":
+    pytest.main()
\ No newline at end of file