import unittest

import lsst.utils.tests

from lsst.skymap.discreteSkyMap import DiscreteSkyMap
from helper import skyMapTestCase


# These are the PS1 Medium-Deep fields
coords = [(10.6750, 41.2667),  # M31
          (36.2074, -04.5833),  # XMM-LSS
          (53.1, -28.1333),  # CDFS
          (130.5917, +44.3167),  # IfA-Lynx
          (150.0, +02.2),  # COSMOS
          (161.9167, +58.0833),  # Lockman
          (185.0, +47.1167),  # NGC4258
          (213.7051, +53.0834),  # DEEP2-Field1/EGS
          (242.7875, +54.95),  # EliasN1
          (334.1875, +00.2833),  # SA22
          (352.3125, -00.4333),  # DEEP2-Field3
          (270.0, +66.56),  # North Ecliptic Pole
          ]
config = DiscreteSkyMap.ConfigClass()
config.raList = [c[0] for c in coords]
config.decList = [c[1] for c in coords]
config.radiusList = [2] * len(coords)


class DiscreteTestCase(skyMapTestCase.SkyMapTestCase):

    def setUp(self):
        self.setAttributes(
            SkyMapClass=DiscreteSkyMap,
            name="discrete",
            config=config,
            numTracts=len(coords),
            neighborAngularSeparation=None,  # don't test for fixed angular sep
            numNeighbors=None,  # ignored because neighborAngularSeparation is None
        )

    def testCompare(self):
        """Test that DiscreteSkyMap's extra state is included in its hash."""
        defaultSkyMap = self.getSkyMap()
        for index in (3, 5):
            config = self.getConfig()
            # delete one tract
            del config.raList[index]
            del config.decList[index]
            del config.radiusList[index]
            skyMap = self.getSkyMap(config=config)
            self.assertNotEqual(skyMap, defaultSkyMap)
        for radius in (1.8, 2.3):
            config = self.getConfig()
            config.radiusList[5] = radius
            skyMap = self.getSkyMap(config=config)
            self.assertNotEqual(skyMap, defaultSkyMap)
        for ra in (35.1, 72.6):
            config = self.getConfig()
            config.raList[5] = ra
            skyMap = self.getSkyMap(config=config)
            self.assertNotEqual(skyMap, defaultSkyMap)
        for dec in (-5.2, 1.8):
            config = self.getConfig()
            config.decList[5] = dec
            skyMap = self.getSkyMap(config=config)
            self.assertNotEqual(skyMap, defaultSkyMap)


class MemoryTester(lsst.utils.tests.MemoryTestCase):
    pass


def setup_module(module):
    lsst.utils.tests.init()


if __name__ == "__main__":
    lsst.utils.tests.init()
    unittest.main()