FWCore/ParameterSet/python/Options.py - master - githubmirror/cmssw

FWCore/ParameterSet/python/Options.py

29 строк785 B

Giulio Eulisse

Snapshot of CMSSW_6_2_0_pre8.

26 июн 2013, 18:12

26 июн 2013, 18:12214ab73

class Options(dict):

    def __init__(self, *args, **kw):
        dict.__init__(self, *args, **kw)
        self.readKeys = set()

    def __getitem__(self, key):
        self.readKeys.add(key)
        return self.get(key,None)

    def _unreadKeys(self):
        """return unused keys"""
        return set([a for a in self if a not in self.readKeys])
    unreadKeys = property(_unreadKeys)


##########################
if __name__ == "__main__":
    import unittest
    class TestOptions(unittest.TestCase):
        def testOptions(self):
            a = Options()
            a["A"] = 3
            a["B"] = 3
            a["A"]
            self.assertEqual(a.readKeys, set(["A"]))
            self.assertEqual(a.unreadKeys, set(["B"]))                        

    unittest.main()