class BigBoolTestCase(unittest.TestCase): def setUp(self): self.policy = Policy() def tearDown(self): del self.policy def testBigBoolArray(self): biglen = 1000 self.policy.isBool("true") for i in range(biglen): self.policy.add("true", True) v = self.policy.getArray("true") self.assertEqual(len(v), biglen, "wrong big number of values in array") for i in range(biglen): self.assertTrue(v[i], "big array with bad value") del v self.assertTrue(True, "Blew up True") fd = open("/dev/null", "w") print("look: %s" % True, file=fd) fd.close()
def testPolicySetget(self): p = Policy() self.assertFalse(p.exists("foo"), "empty existence test failed") self.assertEqual(p.valueCount("foo.bar"), 0, "empty valueCount test failed") self.assertRaises(lsst.pex.exceptions.Exception, p.getTypeInfo, "foo") p.set("doall", "true") # non-existence tests on a non-empty policy self.assertFalse(p.exists("foo"), "non-empty non-existence test failed") self.assertEqual(p.valueCount("foo.bar"), 0, "empty valueCount test failed") self.assertFalse(p.isInt("foo"), "non-empty non-existence type test failed") self.assertRaises(lsst.pex.exceptions.Exception, p.getTypeInfo, "foo") # existence tests self.assertTrue(p.exists("doall"), "non-empty existence test failed") self.assertEqual(p.valueCount("doall"), 1, "single valueCount test failed") self.assertRaises(lsst.pex.exceptions.Exception, p.getInt, "doall") self.assertRaises(lsst.pex.exceptions.Exception, p.getDoubleArray, "doall") self.assertEqual(p.get("doall"), "true", "top-level getString failed") p.set("doall", "duh") self.assertEqual(p.get("doall"), "duh", "top-level getString failed") # test array access ary = p.getArray("doall") self.assertEqual(len(ary), 1, "scalar property has more than one value") self.assertEqual(ary[0], "duh", "scalar access via array failed") p.add("doall", "never") self.assertEqual(p.valueCount("doall"), 2, "2-elem. valueCount test failed") self.assertEqual(p.get("doall"), "never", "top-level add failed") ary = p.getArray("doall") self.assertEqual(len(ary), 2, "scalar property has wrong number of values") self.assertEqual(ary[0], "duh", "scalar access via (2-el) array failed") self.assertEqual(ary[-1], "never", "scalar access via (2-el) array failed") # test hierarchical access # list names # test types p.set("pint", 5) self.assertEqual(p.getInt("pint"), 5, "support for type int failed") self.assertIsInstance(p.get("pint"), int, "auto-typing for int failed") p.set("pdbl", 5.1) self.assertAlmostEqual(p.getDouble("pdbl"), 5.1, 7, "support for type double failed") self.assertEqual(type(p.get("pdbl")), type(5.1), "auto-typing for double failed") p.set("pbool", True) self.assertTrue(p.getBool("pbool"), "support for type bool failed") self.assertEqual(type(p.get("pbool")), type(True), "auto-typing for bool failed") p.add("pbool", False) # test shallow & deep copies # test raise NameNotFound if not present self.assertRaises(NameNotFound, p.get, "nonexistent") self.assertRaises(NameNotFound, p.getArray, "nonexistent") self.assertRaises(NameNotFound, p.getDouble, "nonexistent")
def testPolicySetget(self): p = Policy() self.assert_(not p.exists("foo"), "empty existence test failed") self.assertEqual(p.valueCount("foo.bar"), 0, "empty valueCount test failed") self.assertRaises(NameNotFound, p.getTypeInfo, "foo") p.set("doall", "true") # non-existence tests on a non-empty policy self.failUnless(not p.exists("foo"), "non-empty non-existence test failed") self.assertEqual(p.valueCount("foo.bar"), 0, "empty valueCount test failed") self.failUnless(not p.isInt("foo"), "non-empty non-existence type test failed") self.assertRaises(NameNotFound, p.getTypeInfo, "foo") # existence tests self.assert_(p.exists("doall"), "non-empty existence test failed") self.assertEquals(p.valueCount("doall"), 1, "single valueCount test failed") self.assertRaises(Exception, p.getInt, "doall") self.assertRaises(Exception, p.getDoubleArray, "doall") self.assertEquals(p.get("doall"), "true", "top-level getString failed") p.set("doall", "duh") self.assertEquals(p.get("doall"), "duh", "top-level getString failed") # test array access ary = p.getArray("doall") self.assertEquals(len(ary), 1, "scalar property has more than one value") self.assertEquals(ary[0], "duh", "scalar access via array failed") p.add("doall", "never") self.assertEquals(p.valueCount("doall"), 2, "2-elem. valueCount test failed") self.assertEquals(p.get("doall"), "never", "top-level add failed") ary = p.getArray("doall") self.assertEquals(len(ary), 2, "scalar property has wrong number of values") self.assertEquals(ary[0], "duh", "scalar access via (2-el) array failed") self.assertEquals(ary[-1], "never", "scalar access via (2-el) array failed") # test hierarchical access # list names # test types p.set("pint", 5) self.assertEquals(p.getInt("pint"), 5, "support for type int failed") self.assertEquals(type(p.get("pint")), type(5), "auto-typing for int failed") p.set("pdbl", 5.1) self.assertAlmostEquals(p.getDouble("pdbl"), 5.1, 7, "support for type double failed") self.assertEquals(type(p.get("pdbl")), type(5.1), "auto-typing for double failed") p.set("pbool", True) self.assert_(p.getBool("pbool"), "support for type bool failed") self.assertEquals(type(p.get("pbool")), type(True), "auto-typing for bool failed") p.add("pbool", False) # test shallow & deep copies # test raise NameNotFound if not present try: p.get("nonexistent") self.fail() # should never reach here except Exception, e: self.assert_(isinstance(e, NameNotFound))
class PolicyBlackboardItem(BlackboardItem): """ An implementation of a BlackboardItem that stores properities via a policy """ def __init__(self, policyfile=None): """ create an item with the given properties. @param policyfile A policy """ # the properties attached to this items self._props = None if policyfile: self._props = Policy.createPolicy(policyfile) else: self._props = Policy() def getProperty(self, name, default=None): """ return the value for a property @param name the property name @parma default the default value to return if the name is not set """ if not self._props.exists(name): return default elif self._props.isArray(name): return self._props.getArray(name) else: return self._props.get(name) def hasProperty(self, name): """ return True if the property with the given name is available """ return self._props.exists(name) def __getitem__(self, name): if not self._props.exists(name): raise KeyError(name) return BlackboardItem.__getitem__(self, name) def _setProperty(self, name, val): # set a property value if isinstance(val, list): self._props.set(name, val.pop(0)) for v in val: self._props.add(name, v) else: self._props.set(name, val) def getPropertyNames(self): """ return the property names that make up this item """ return self._props.names() def _copyFrom(self, item): for name in item.getPropertyNames(): self._setProperty(name, item.getProperty(name)) @staticmethod def createFormatter(): return PolicyBlackboardItem._Fmtr() class _Fmtr(object): def write(self, filename, item): pol = None if isinstance(item, PolicyBlackboardItem): pol = item._props else: delegate = PolicyBlackboardItem() delegate._copyFrom(item) pol = delegate._props writer = PAFWriter(filename) try: writer.write(pol, True) finally: writer.close() def openItem(self, filename): out = PolicyBlackboardItem() out._props = Policy.createPolicy(filename) return out def filenameExt(self): """ return the recommended extension for the format this writes out """ return "paf"