consolidate maskbits tests

pydl/pydlutils/sdss/tests/test_maskbits.py

@@ -4,7 +4,7 @@
 from os.path import dirname, join
 import pydl.pydlutils.sdss
 from ..set_maskbits import set_maskbits
-from .. import sdss_flagname, sdss_flagval
+from .. import sdss_flagexist, sdss_flagname, sdss_flagval
 
 
 class TestMaskbits(object):
@@ -19,6 +19,24 @@ def setup(self):
     def teardown(self):
         pass
 
+    def test_sdss_flagexist(self):
+        assert sdss_flagexist('TARGET', 'ROSAT_A')
+        assert sdss_flagexist('TARGET',
+                            ['rosat_a', 'rosat_b', 'rosat_c', 'rosat_d'])
+        l, f = sdss_flagexist('target', 'FOOBAR', flagexist=True)
+        assert not l
+        assert f
+        l, which = sdss_flagexist('TARGET', ['rosat_a', 'rosat_b', 'rosat_c',
+                                'rosat_d', 'foobar'], whichexist=True)
+        assert not l
+        assert tuple(which) == (True, True, True, True, False)
+        l, f, which = sdss_flagexist('TARGET', ['rosat_a', 'rosat_b',
+                                    'rosat_c','rosat_d','foobar'],
+                                    flagexist=True, whichexist=True)
+        assert not l
+        assert f
+        assert tuple(which) == (True, True, True, True, False)
+
     def test_sdss_flagname(self):
         names = sdss_flagname('ANCILLARY_TARGET1', 2310346608843161600)
         assert tuple(names) == ('BRIGHTGAL', 'BLAZGX', 'ELG')
@@ -37,3 +55,20 @@ def test_sdss_flagval(self):
             val = sdss_flagval('TARGET', 'ROSAT_Q')
         with raises(KeyError):
             val = sdss_flagval('ABADMASK', "ABADFLAG")
+
+    def test_set_maskbits(self):
+        maskbits = pydl.pydlutils.sdss.maskbits
+        assert (set(maskbits.keys()) ==
+            set(['TARGET', 'BOSS_TARGET1', 'PRIMTARGET', 'ANCILLARY_TARGET1',
+                'ZWARNING', 'TTARGET', 'SECTARGET', 'LEGACY_TARGET2',
+                'LEGACY_TARGET1', 'SPECIAL_TARGET2', 'FLUXMATCH_STATUS']))
+        assert (set(maskbits['TARGET'].keys()) ==
+            set(['QSO_FIRST_SKIRT', 'QSO_CAP', 'GALAXY_RED', 'STAR_CARBON',
+                'STAR_WHITE_DWARF', 'GALAXY_RED_II', 'GALAXY_BIG',
+                'GALAXY_BRIGHT_CORE', 'SERENDIP_MANUAL', 'STAR_SUB_DWARF',
+                'QSO_FIRST_CAP', 'QSO_SKIRT', 'STAR_PN', 'STAR_BHB', 'QSO_HIZ',
+                'STAR_BROWN_DWARF', 'SERENDIP_FIRST', 'SOUTHERN_SURVEY',
+                'STAR_RED_DWARF', 'STAR_CATY_VAR', 'QSO_REJECT', 'GALAXY',
+                'SERENDIP_RED', 'SERENDIP_DISTANT', 'QSO_MAG_OUTLIER',
+                'ROSAT_A', 'ROSAT_C', 'ROSAT_B', 'ROSAT_E', 'ROSAT_D',
+                'SERENDIP_BLUE']))