Unverified Commit d3298ca0 authored by Akke Viitanen's avatar Akke Viitanen
Browse files

fix missing import

parent e90ff718

tests/lsst_inaf_agile/test_catalog_combined.py

+6 −4
Original line number Diff line number Diff line
@@ -9,6 +9,7 @@ import os 
from unittest import TestCase



import numpy as np

from lsst_inaf_agile import util

from lsst_inaf_agile.catalog_agn import CatalogAGN

from lsst_inaf_agile.catalog_combined import CatalogCombined

from lsst_inaf_agile.catalog_galaxy import CatalogGalaxy
@@ -40,6 +41,11 @@ def create_combined_catalog(dirname="data/tests/test_catalog_combined"): 
    catalog_star = CatalogStar(dirname, catalog_galaxy, is_binary=False)

    catalog_binary = CatalogStar(dirname, catalog_galaxy, is_binary=True)



    # NOTE: achieve fbin=0.4, see dal Tio+ Sec. 3.2

    catalog_star.stars, catalog_binary.stars = util.get_star_binary_fbin(

        catalog_star.stars, catalog_binary.stars, fbin=0.40, nrepeat=4

    )



    # Create the combined catalog

    catalog_combined = CatalogCombined(dirname, catalog_galaxy, catalog_agn, catalog_star, catalog_binary)
@@ -201,10 +207,6 @@ class TestCatalogCombined(TestCase): 
            os.remove(filename)

        self.catalog_combined.write()

        self.assertTrue(os.path.exists(filename))



        # read the file

        from lsst_inaf_agile import util



        fits = util.read_fits(filename)

        self.assertGreater(len(fits), 0)