def setUp(self):
self.key = "basic"
config = """
import shipunit as u
from ShipGeoConfig import AttrDict, ConfigRegistry
with ConfigRegistry.register_config("basic") as c:
c.vetoStation = AttrDict(z=-2390.*u.cm)
c.TrackStation1 = AttrDict(z=1510.*u.cm)
c.TrackStation2 = AttrDict(z=1710.*u.cm)
c.TrackStation3 = AttrDict(z=2150.*u.cm)
c.TrackStation4 = AttrDict(z=2370.*u.cm)
c.z = c.TrackStation2.z + 0.5 * (c.TrackStation3.z - c.TrackStation2.z)
c.Bfield = AttrDict(z=c.z)
c.Bfield.max = 1.5*u.kilogauss # was 1.15 in EOI
# target absorber muon shield setup
c.decayVolume = AttrDict(z=0*u.cm)
c.decayVolume.length = 50*u.m
c.muShield = AttrDict(z=0*u.cm)
c.muShield.dZ1 = 2.5*u.m
c.muShield.dZ2 = 3.5*u.m
c.muShield.dZ3 = 3.0*u.m
c.muShield.dZ4 = 3.0*u.m
c.muShield.dZ5 = 2.5*u.m
c.muShield.dZ6 = 2.5*u.m
c.muShield.LE = 5*u.m
"""
ConfigRegistry.loadpys(config)
def setUp(self):
self.key = "basic"
config = """
import shipunit as u
from ShipGeoConfig import AttrDict, ConfigRegistry
with ConfigRegistry.register_config("basic") as c:
c.vetoStation = AttrDict(z=-2390.*u.cm)
c.TrackStation1 = AttrDict(z=1510.*u.cm)
c.TrackStation2 = AttrDict(z=1710.*u.cm)
c.TrackStation3 = AttrDict(z=2150.*u.cm)
c.TrackStation4 = AttrDict(z=2370.*u.cm)
c.z = c.TrackStation2.z + 0.5 * (c.TrackStation3.z - c.TrackStation2.z)
c.Bfield = AttrDict(z=c.z)
c.Bfield.max = 1.5*u.kilogauss # was 1.15 in EOI
# target absorber muon shield setup
c.decayVolume = AttrDict(z=0*u.cm)
c.decayVolume.length = 50*u.m
c.muShield = AttrDict(z=0*u.cm)
c.muShield.dZ1 = 2.5*u.m
c.muShield.dZ2 = 3.5*u.m
c.muShield.dZ3 = 3.0*u.m
c.muShield.dZ4 = 3.0*u.m
c.muShield.dZ5 = 2.5*u.m
c.muShield.dZ6 = 2.5*u.m
c.muShield.LE = 5*u.m
"""
ConfigRegistry.loadpys(config)
def test_true(self):
c = ConfigRegistry.loadpys(self.config, MU_SHIELD_ENABLED=True)
self.assertTrue("muShield" in c)
assert len(ConfigRegistry.keys()) == 1, ConfigRegistry.keys()
assert self.key in ConfigRegistry.keys()
assert ConfigRegistry[self.key].Bfield.max == 1.5*u.kilogauss
assert ConfigRegistry[self.key].muShield.dZ1 == 2.5*u.m
def test_true(self):
c = ConfigRegistry.loadpys(self.config, MU_SHIELD_ENABLED=True)
self.assertIn("muShield", c)
assert len(ConfigRegistry.keys()) == 1, ConfigRegistry.keys()
assert self.key in ConfigRegistry.keys()
assert ConfigRegistry[self.key].Bfield.max == 1.5 * u.kilogauss
assert ConfigRegistry[self.key].muShield.dZ1 == 2.5 * u.m
def test_true(self):
c = ConfigRegistry.loadpys(self.config)
self.assertTrue("vetoStation" in c)
assert ConfigRegistry[self.key].vetoStation.z == -2390*u.cm
def test_false(self):
ConfigRegistry.loadpys(self.config, MU_SHIELD_ENABLED=False)
assert len(ConfigRegistry.keys()) == 1, ConfigRegistry.keys()
assert self.key in ConfigRegistry.keys()
assert ConfigRegistry[self.key].Bfield.max == 1.5*u.kilogauss
self.assertTrue("muShield" not in ConfigRegistry[self.key])
def test_false(self):
ConfigRegistry.loadpys(self.config, MU_SHIELD_ENABLED=False)
assert len(ConfigRegistry.keys()) == 1, ConfigRegistry.keys()
assert self.key in ConfigRegistry.keys()
assert ConfigRegistry[self.key].Bfield.max == 1.5 * u.kilogauss
self.assertNotIn("muShield", ConfigRegistry[self.key])
def test_true(self):
c = ConfigRegistry.loadpys(self.config)
self.assertTrue("vetoStation" in c)
assert ConfigRegistry[self.key].vetoStation.z == -2390 * u.cm
