3179 def testModifier(self):
3181 p = Process(
"test",m1)
3182 p.a = EDAnalyzer(
"MyAnalyzer", fred = int32(1))
3185 m1.toModify(p.a,_mod_fred)
3186 self.assertEqual(p.a.fred.value(),2)
3187 p.b = EDAnalyzer(
"YourAnalyzer", wilma = int32(1))
3188 m1.toModify(p.b, wilma = 2)
3189 self.assertEqual(p.b.wilma.value(),2)
3190 self.assertTrue(p.isUsingModifier(m1))
3194 p.a = EDAnalyzer(
"MyAnalyzer", fred = int32(1))
3195 m1.toModify(p.a,_mod_fred)
3196 p.b = EDAnalyzer(
"YourAnalyzer", wilma = int32(1))
3197 m1.toModify(p.b, wilma = 2)
3198 self.assertEqual(p.a.fred.value(),1)
3199 self.assertEqual(p.b.wilma.value(),1)
3200 self.assertEqual(p.isUsingModifier(m1),
False)
3203 p = Process(
"test",m1)
3204 p.a = EDAnalyzer(
"MyAnalyzer", fred = int32(1), wilma = int32(1))
3205 m1.toModify(p.a, fred = int32(2))
3206 p.b = p.a.clone(wilma = int32(3))
3207 self.assertEqual(p.a.fred.value(),2)
3208 self.assertEqual(p.a.wilma.value(),1)
3209 self.assertEqual(p.b.fred.value(),2)
3210 self.assertEqual(p.b.wilma.value(),3)
3213 p = Process(
"test",m1)
3214 p.a = EDAnalyzer(
"MyAnalyzer", fred = int32(1), wilma = int32(1), fintstones = PSet(fred = int32(1)))
3215 m1.toModify(p.a, fred =
None, fintstones = dict(fred =
None))
3216 self.assertEqual(hasattr(p.a,
"fred"),
False)
3217 self.assertEqual(hasattr(p.a.fintstones,
"fred"),
False)
3218 self.assertEqual(p.a.wilma.value(),1)
3221 p = Process(
"test",m1)
3222 p.a = EDAnalyzer(
"MyAnalyzer", fred = int32(1))
3223 m1.toModify(p.a, wilma = int32(2))
3224 self.assertEqual(p.a.fred.value(), 1)
3225 self.assertEqual(p.a.wilma.value(),2)
3228 p = Process(
"test",m1)
3229 p.a = EDAnalyzer(
"MyAnalyzer", flintstones = PSet(fred = int32(1), wilma = int32(1)))
3230 m1.toModify(p.a, flintstones = dict(fred = int32(2)))
3231 self.assertEqual(p.a.flintstones.fred.value(),2)
3232 self.assertEqual(p.a.flintstones.wilma.value(),1)
3235 p = Process(
"test",m1)
3236 p.a = EDAnalyzer(
"MyAnalyzer", flintstones = PSet(fred = PSet(wilma = int32(1))))
3237 self.assertRaises(KeyError,
lambda: m1.toModify(p.a, flintstones = dict(imnothere = dict(wilma=2))))
3238 self.assertRaises(KeyError,
lambda: m1.toModify(p.a, foo = 1))
3241 p = Process(
"test",m1)
3242 p.a = EDAnalyzer(
"MyAnalyzer", flintstones =
VPSet(PSet(fred = int32(1)), PSet(wilma = int32(1))))
3243 m1.toModify(p.a, flintstones = {1:dict(wilma = int32(2))})
3244 self.assertEqual(p.a.flintstones[0].fred.value(),1)
3245 self.assertEqual(p.a.flintstones[1].wilma.value(),2)
3248 p = Process(
"test",m1)
3249 p.a = EDAnalyzer(
"MyAnalyzer", fred = vuint32(1,2,3))
3250 m1.toModify(p.a, fred = {1:7})
3251 self.assertEqual(p.a.fred[0],1)
3252 self.assertEqual(p.a.fred[1],7)
3253 self.assertEqual(p.a.fred[2],3)
3256 p = Process(
"test",m1)
3257 p.a = EDAnalyzer(
"MyAnalyzer", fred = vuint32(1,2,3))
3259 try: m1.toModify(p.a, fred = {5:7})
3260 except IndexError
as e: raised =
True
3261 self.assertEqual(raised,
True)
3264 p = Process(
"test",m1)
3265 p.a = EDAnalyzer(
"MyAnalyzer", flintstones =
VPSet(PSet(fred = int32(1)), PSet(wilma = int32(1))))
3267 try: m1.toModify(p.a, flintstones = dict(bogus = int32(37)))
3268 except TypeError
as e: raised =
True
3269 self.assertEqual(raised,
True)
3273 class ProcModifierMod(
object):
3274 def __init__(self,modifier,func):
3275 self.proc_mod_ = modifier.makeProcessModifier(func)
3278 self.a = EDAnalyzer(
"Dummy")
3279 testMod = DummyMod()
3281 self.assertTrue(hasattr(p,
"a"))
3283 p = Process(
"test",m1)
3284 testProcMod = ProcModifierMod(m1,_rem_a)
3286 p.extend(testProcMod)
3287 self.assertTrue(
not hasattr(p,
"a"))
3290 mc = ModifierChain(m1)
3291 p = Process(
"test",mc)
3292 self.assertTrue(p.isUsingModifier(m1))
3293 self.assertTrue(p.isUsingModifier(mc))
3294 testMod = DummyMod()
3295 p.b = EDAnalyzer(
"Dummy2", fred = int32(1))
3296 m1.toModify(p.b, fred = int32(3))
3298 testProcMod = ProcModifierMod(m1,_rem_a)
3299 p.extend(testProcMod)
3300 self.assertTrue(
not hasattr(p,
"a"))
3301 self.assertEqual(p.b.fred.value(),3)
3305 mc = ModifierChain(m1,m2)
3306 mclone = mc.copyAndExclude([m2])
3307 self.assertTrue(
not mclone._isOrContains(m2))
3308 self.assertTrue(mclone._isOrContains(m1))
3310 mc2 = ModifierChain(mc,m3)
3311 mclone = mc2.copyAndExclude([m2])
3312 self.assertTrue(
not mclone._isOrContains(m2))
3313 self.assertTrue(mclone._isOrContains(m1))
3314 self.assertTrue(mclone._isOrContains(m3))
3318 p = Process(
"test",m1)
3319 p.a = EDAnalyzer(
"MyAnalyzer", fred = int32(1), wilma = int32(1))
3320 (m1 & m2).toModify(p.a, fred = int32(2))
3321 self.assertRaises(TypeError,
lambda: (m1 & m2).toModify(p.a, 1, wilma=2))
3322 self.assertEqual(p.a.fred, 1)
3325 p = Process(
"test",m1,m2)
3326 p.a = EDAnalyzer(
"MyAnalyzer", fred = int32(1), wilma = int32(1))
3327 (m1 & m2).toModify(p.a, fred = int32(2))
3328 self.assertEqual(p.a.fred, 2)
3332 p = Process(
"test",m1,m2,m3)
3333 p.a = EDAnalyzer(
"MyAnalyzer", fred = int32(1), wilma = int32(1))
3334 (m1 & m2 & m3).toModify(p.a, fred = int32(2))
3335 self.assertEqual(p.a.fred, 2)
3336 (m1 & (m2 & m3)).toModify(p.a, fred = int32(3))
3337 self.assertEqual(p.a.fred, 3)
3338 ((m1 & m2) & m3).toModify(p.a, fred = int32(4))
3339 self.assertEqual(p.a.fred, 4)
3343 p = Process(
"test", m1)
3344 p.a = EDAnalyzer(
"MyAnalyzer", fred = int32(1), wilma = int32(1))
3345 (~m1).toModify(p.a, fred=2)
3346 self.assertEqual(p.a.fred, 1)
3347 (~m2).toModify(p.a, wilma=2)
3348 self.assertEqual(p.a.wilma, 2)
3349 self.assertRaises(TypeError,
lambda: (~m1).toModify(p.a, 1, wilma=2))
3350 self.assertRaises(TypeError,
lambda: (~m2).toModify(p.a, 1, wilma=2))
3355 p = Process(
"test", m1)
3356 p.a = EDAnalyzer(
"MyAnalyzer", fred = int32(1), wilma = int32(1))
3357 (m1 | m2).toModify(p.a, fred=2)
3358 self.assertEqual(p.a.fred, 2)
3359 (m1 | m2 | m3).toModify(p.a, fred=3)
3360 self.assertEqual(p.a.fred, 3)
3361 (m3 | m2 | m1).toModify(p.a, fred=4)
3362 self.assertEqual(p.a.fred, 4)
3363 ((m1 | m2) | m3).toModify(p.a, fred=5)
3364 self.assertEqual(p.a.fred, 5)
3365 (m1 | (m2 | m3)).toModify(p.a, fred=6)
3366 self.assertEqual(p.a.fred, 6)
3367 (m2 | m3).toModify(p.a, fred=7)
3368 self.assertEqual(p.a.fred, 6)
3369 self.assertRaises(TypeError,
lambda: (m1 | m2).toModify(p.a, 1, wilma=2))
3370 self.assertRaises(TypeError,
lambda: (m2 | m3).toModify(p.a, 1, wilma=2))
3376 p = Process(
"test", m1, m2)
3377 p.a = EDAnalyzer(
"MyAnalyzer", fred = int32(1), wilma = int32(1))
3378 (m1 & ~m2).toModify(p.a, fred=2)
3379 self.assertEqual(p.a.fred, 1)
3380 (m1 & ~m3).toModify(p.a, fred=2)
3381 self.assertEqual(p.a.fred, 2)
3382 (m1 | ~m2).toModify(p.a, fred=3)
3383 self.assertEqual(p.a.fred, 3)
3384 (~m1 | ~m2).toModify(p.a, fred=4)
3385 self.assertEqual(p.a.fred, 3)
3386 (~m3 & ~m4).toModify(p.a, fred=4)
3387 self.assertEqual(p.a.fred, 4)
3388 ((m1 & m3) | ~m4).toModify(p.a, fred=5)
3389 self.assertEqual(p.a.fred, 5)
3392 p = Process(
"test",m1)
3393 p.a =EDAnalyzer(
"MyAnalyzer", fred = int32(1))
3394 m1.toReplaceWith(p.a, EDAnalyzer(
"YourAnalyzer", wilma = int32(3)))
3395 self.assertRaises(TypeError,
lambda: m1.toReplaceWith(p.a, EDProducer(
"YourProducer")))
3396 p.b =EDAnalyzer(
"BAn")
3397 p.c =EDProducer(
"c")
3398 p.d =EDProducer(
"d")
3402 m1.toReplaceWith(p.s,
Sequence(p.a+p.b, p.td))
3403 self.assertEqual(p.a.wilma.value(),3)
3404 self.assertEqual(p.a.type_(),
"YourAnalyzer")
3405 self.assertEqual(hasattr(p,
"fred"),
False)
3406 self.assertTrue(p.s.dumpPython() ==
"cms.Sequence(process.a+process.b, process.td)\n")
3407 p.e =EDProducer(
"e")
3408 m1.toReplaceWith(p.td, Task(p.e))
3409 self.assertTrue(p.td._collection ==
OrderedSet([p.e]))
3413 p.a =EDAnalyzer(
"MyAnalyzer", fred = int32(1))
3414 m1.toReplaceWith(p.a, EDAnalyzer(
"YourAnalyzer", wilma = int32(3)))
3415 self.assertEqual(p.a.type_(),
"MyAnalyzer")
3421 p = Process(
"test", m1, m2)
3422 p.a = EDAnalyzer(
"MyAnalyzer", fred = int32(1), wilma = int32(1))
3423 self.assertRaises(TypeError,
lambda: (m1 & m2).toReplaceWith(p.a, EDProducer(
"YourProducer")))
3424 self.assertRaises(TypeError,
lambda: (m3 & m4).toReplaceWith(p.a, EDProducer(
"YourProducer")))
3425 self.assertRaises(TypeError,
lambda: (~m3).toReplaceWith(p.a, EDProducer(
"YourProducer")))
3426 self.assertRaises(TypeError,
lambda: (~m1).toReplaceWith(p.a, EDProducer(
"YourProducer")))
3427 self.assertRaises(TypeError,
lambda: (m1 | m3).toReplaceWith(p.a, EDProducer(
"YourProducer")))
3428 self.assertRaises(TypeError,
lambda: (m3 | m4).toReplaceWith(p.a, EDProducer(
"YourProducer")))
3429 (m1 & m2).toReplaceWith(p.a, EDAnalyzer(
"YourAnalyzer1"))
3430 self.assertEqual(p.a.type_(),
"YourAnalyzer1")
3431 (m1 & m3).toReplaceWith(p.a, EDAnalyzer(
"YourAnalyzer2"))
3432 self.assertEqual(p.a.type_(),
"YourAnalyzer1")
3433 (~m1).toReplaceWith(p.a, EDAnalyzer(
"YourAnalyzer2"))
3434 self.assertEqual(p.a.type_(),
"YourAnalyzer1")
3435 (~m3).toReplaceWith(p.a, EDAnalyzer(
"YourAnalyzer2"))
3436 self.assertEqual(p.a.type_(),
"YourAnalyzer2")
3437 (m1 | m3).toReplaceWith(p.a, EDAnalyzer(
"YourAnalyzer3"))
3438 self.assertEqual(p.a.type_(),
"YourAnalyzer3")
3439 (m3 | m4).toReplaceWith(p.a, EDAnalyzer(
"YourAnalyzer4"))
3440 self.assertEqual(p.a.type_(),
"YourAnalyzer3")
3443 a = EDAlias(foo2 =
VPSet(PSet(type =
string(
"Foo2"))))
3447 m.toModify(a, foo2 = {0: dict(type =
"Foo3")})
3448 self.assertEqual(a.foo2[0].type,
"Foo3")
3450 m.toModify(a, foo4 =
VPSet(PSet(type =
string(
"Foo4"))))
3451 self.assertEqual(a.foo2[0].type,
"Foo3")
3452 self.assertEqual(a.foo4[0].type,
"Foo4")
3454 m.toModify(a, foo2 =
None)
3455 self.assertFalse(hasattr(a,
"foo2"))
3456 self.assertEqual(a.foo4[0].type,
"Foo4")
3458 m.toReplaceWith(a, EDAlias(bar =
VPSet(PSet(type =
string(
"Bar")))))
3459 self.assertFalse(hasattr(a,
"foo2"))
3460 self.assertFalse(hasattr(a,
"foo4"))
3461 self.assertTrue(hasattr(a,
"bar"))
3462 self.assertEqual(a.bar[0].type,
"Bar")
3465 sp = SwitchProducerTest(test1 = EDProducer(
"Foo",
3467 b = PSet(c = int32(2))),
3468 test2 = EDProducer(
"Bar",
3470 bb = PSet(cc = int32(12))))
3475 test1 = dict(a = 4, b = dict(c =
None)),
3476 test2 = dict(aa = 15, bb = dict(cc = 45, dd =
string(
"foo"))))
3477 self.assertEqual(sp.test1.a.value(), 4)
3478 self.assertEqual(sp.test1.b.hasParameter(
"c"),
False)
3479 self.assertEqual(sp.test2.aa.value(), 15)
3480 self.assertEqual(sp.test2.bb.cc.value(), 45)
3481 self.assertEqual(sp.test2.bb.dd.value(),
"foo")
3483 m.toReplaceWith(sp.test1, EDProducer(
"Fred", x = int32(42)))
3484 self.assertEqual(sp.test1.type_(),
"Fred")
3485 self.assertEqual(sp.test1.x.value(), 42)
3486 self.assertRaises(TypeError,
lambda: m.toReplaceWith(sp.test1, EDAnalyzer(
"Foo")))
3488 m.toModify(sp, test2 = EDProducer(
"Xyzzy", x = int32(24)))
3489 self.assertEqual(sp.test2.type_(),
"Xyzzy")
3490 self.assertEqual(sp.test2.x.value(), 24)
3491 self.assertRaises(TypeError,
lambda: m.toModify(sp, test2 = EDAnalyzer(
"Foo")))
3493 m.toModify(sp, test3 = EDProducer(
"Wilma", y = int32(24)))
3494 self.assertEqual(sp.test3.type_(),
"Wilma")
3495 self.assertEqual(sp.test3.y.value(), 24)
3496 self.assertRaises(TypeError,
lambda: m.toModify(sp, test4 = EDAnalyzer(
"Foo")))
3498 m.toModify(sp, test2 =
None)
3499 self.assertEqual(hasattr(sp,
"test2"),
False)
3501 m.toModify(sp, test2 = EDAlias(foo =
VPSet(PSet(type =
string(
"int")))))
3502 self.assertTrue(hasattr(sp.test2,
"foo"))
3504 m.toReplaceWith(sp.test2, EDAlias(bar =
VPSet(PSet(type =
string(
"int")))))
3505 self.assertTrue(hasattr(sp.test2,
"bar"))
3507 m.toModify(sp, test2 = EDAlias(xyzzy =
VPSet(PSet(type =
string(
"int")))))
3508 self.assertTrue(hasattr(sp.test2,
"xyzzy"))
3510 self.assertRaises(TypeError,
lambda: m.toReplaceWith(sp.test2, EDProducer(
"Foo")))
3511 m.toModify(sp, test2 = EDProducer(
"Foo"))