def setUp(self):
super(TestEggResource, self).setUp()
cwd = os.getcwd()
os.chdir(self.base)
with captured_output('stdout'):
with captured_output('stderr'):
setuptools.setup(
name=self.mod,
packages=[self.mod],
script_args=['bdist_egg', '--quiet'],
package_data={
self.mod: ['a.yml',
os.path.join('subdir', 'b.yml')]
})
os.chdir(cwd)
self.eggs = os.listdir(os.path.join(self.base, 'dist'))
for egg in self.eggs:
sys.path.append(os.path.join(self.base, 'dist', egg))
self.res = PackageResource(self.mod)
.. moduleauthor:: Yngve Inntjore Levinsen <Yngve.Inntjore.Levinsen[at]cern.ch>
'''
from __future__ import absolute_import
import logging
from .model import Model
from .model_locator import ChainModelLocator
# Create a default ModelLocator:
# NOTE: we could (and probably should) stop using this global object and
# use only CpymadService objects instead!
from .model_locator import MergedModelLocator
from cern.resource.package import PackageResource
_locator = MergedModelLocator(PackageResource(__package__, '_models'))
default_model_locator = ChainModelLocator()
default_model_locator.add_locator(_locator)
class CpymadService(object):
''' The CPymad implementation of the
abstract class PyMadService. '''
def __init__(self,
model_locator=default_model_locator,
logger=None,
**kwargs):
self._log = logger or logging.getLogger(__name__)
self._am = None
self._models = []
def setUp(self):
super(TestPackageResource, self).setUp()
sys.path.append(self.base)
self.res = PackageResource(self.mod)
