Unverified Commit 6fdb6add authored by Akke Viitanen's avatar Akke Viitanen
Browse files

add test_catalog_combined

parent dc322a0e
Loading
Loading
Loading
Loading
+0 −0

Empty file added.

+66 −0
Original line number Diff line number Diff line
#!/usr/bin/env python3
# Author: Akke Viitanen
# Email: akke.viitanen@helsinki.fi
# Date: 2025-07-03 19:33:15

"""
Tests the galaxy catalog
"""

import logging
import os

from catalog_agn import CatalogAGN
from catalog_combined import CatalogCombined
from catalog_galaxy import CatalogGalaxy
from catalog_star import CatalogStar
from egg import Egg
from image_simulator import ImageSimulator
from merloni2014 import Merloni2014

logger = logging.getLogger(__name__)


def main():
    dirname = "data/catalog/test"
    os.makedirs(dirname, exist_ok=True)

    # Initialize EGG
    egg = Egg.run_config("etc/config_egg.ini")

    # Create the individual catalogs
    catalog_galaxy = CatalogGalaxy(dirname, egg)
    catalog_agn = CatalogAGN(
        dirname=dirname,
        catalog_galaxy=catalog_galaxy,
        type_plambda="zou+2024",
        save_sed=1,
        seed=20250621,
        merloni2014=Merloni2014(1, 0, 0.05, 0.95),
        filter_db="data/egg/share/filter-db/db.dat",
    )
    catalog_star = CatalogStar(dirname, catalog_galaxy, is_binary=False)
    catalog_binary = CatalogStar(dirname, catalog_galaxy, is_binary=True)

    # Create the combined catalog
    catalog_combined = CatalogCombined(dirname, catalog_galaxy, catalog_agn, catalog_star, catalog_binary)

    # Create the reference catalog
    maglim = 24
    selection_band = "lsst-r"
    catalog_combined.write_reference_catalog(f"{dirname}/reference_catalog.csv", maglim, selection_band)

    # Simulate images
    image_simulator = ImageSimulator(f"{dirname}/imsim", catalog_combined, "data/baseline_v4.0_10yrs.db")

    # NOTE: select the first visit from COSMOS DDF
    visits = image_simulator.get_visit()
    select = visits["target_name"] == "DD:COSMOS"
    observation_id = visits["observationId"][select].iloc[0]
    ra_dec = +150.11916667, +2.20583333
    image_simulator.write_instance_catalog(observation_id, ra_dec=ra_dec)
    image_simulator.simulate_image(observation_id, detector=94)


if __name__ == "__main__":
    main()
+13 −0
Original line number Diff line number Diff line
from lsst_inaf_agile import example_module


def test_greetings() -> None:
    """Verify the output of the `greetings` function"""
    output = example_module.greetings()
    assert output == "Hello from LINCC-Frameworks!"


def test_meaning() -> None:
    """Verify the output of the `meaning` function"""
    output = example_module.meaning()
    assert output == 42