/
githubmirror
/
cmssw
Обзор
Документация
Войти
/
githubmirror
/
cmssw
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
master
FWCore/ParameterSet/python/Mixins.py
1 022 строки
42 KB
Chris Jones
Extended unit test to test new None case
14 фев 2026, 00:51
14 фев 2026, 00:51
2a35601
Код
Авторство
О чём код?
from builtins import range, object import inspect import sys from typing import Union class _ConfigureComponent(object): """Denotes a class that can be used by the Processes class""" def _isTaskComponent(self) -> bool: return False class PrintOptions(object): def __init__(self, indent:int = 0, deltaIndent:int = 4, process:bool = True, targetDirectory: Union[str, None] = None, useSubdirectories:bool = False): self.indent_= indent self.deltaIndent_ = deltaIndent self.isCfg = process self.targetDirectory = targetDirectory self.useSubdirectories = useSubdirectories def indentation(self) -> str: return ' '*self.indent_ def indent(self): self.indent_ += self.deltaIndent_ def unindent(self): self.indent_ -= self.deltaIndent_ class _SpecialImportRegistry(object): """This class collects special import statements of configuration types""" def __init__(self): self._registry = {} def _reset(self): for lst in self._registry.values(): lst[1] = False def registerSpecialImportForType(self, cls, impStatement): className = cls.__name__ if className in self._registry: raise RuntimeError("Error: the configuration type '%s' already has an import statement registered '%s'" % (className, self._registry[className][0])) self._registry[className] = [impStatement, False] def registerUse(self, obj): className = obj.__class__.__name__ try: self._registry[className][1] = True except KeyError: pass def getSpecialImports(self): coll = set() for (imp, used) in self._registry.values(): if used: coll.add(imp) return sorted(coll) specialImportRegistry = _SpecialImportRegistry() class _ParameterTypeBase(object): """base class for classes which are used as the 'parameters' for a ParameterSet""" def __init__(self): self.__dict__["_isFrozen"] = False self.__isTracked = True self._isModified = False def isModified(self) -> bool: return self._isModified def resetModified(self): self._isModified=False def configTypeName(self) -> str: if self.isTracked(): return type(self).__name__ return 'untracked '+type(self).__name__ def pythonTypeName(self) -> str: if self.isTracked(): return 'cms.'+type(self).__name__ return 'cms.untracked.'+type(self).__name__ def dumpPython(self, options:PrintOptions=PrintOptions()) -> str: specialImportRegistry.registerUse(self) return self.pythonTypeName()+"("+self.pythonValue(options)+")" def __repr__(self) -> str: return self.dumpPython() def isTracked(self) -> bool: return self.__isTracked def setIsTracked(self,trackness:bool): self.__isTracked = trackness def isFrozen(self) -> bool: return self._isFrozen def setIsFrozen(self): self._isFrozen = True def isCompatibleCMSType(self,aType) -> bool: return isinstance(self,aType) def _checkAndReturnValueWithType(self, valueWithType): if isinstance(valueWithType, type(self)): return valueWithType raise TypeError("Attempted to assign type {from_} to type {to}".format(from_ = str(type(valueWithType)), to = str(type(self))) ) class _SimpleParameterTypeBase(_ParameterTypeBase): """base class for parameter classes which only hold a single value""" def __init__(self,value): super(_SimpleParameterTypeBase,self).__init__() self._value = value if not self._isValid(value): raise ValueError(str(value)+" is not a valid "+str(type(self))) def value(self): return self._value def setValue(self,value): if not self._isValid(value): raise ValueError(str(value)+" is not a valid "+str(type(self))) if value!=self._value: self._isModified=True self._value=value def configValue(self, options:PrintOptions=PrintOptions()) -> str: return str(self._value) def pythonValue(self, options:PrintOptions=PrintOptions()) -> str: return self.configValue(options) def __eq__(self,other) -> bool: if isinstance(other,_SimpleParameterTypeBase): return self._value == other._value return self._value == other def __ne__(self,other) -> bool: if isinstance(other,_SimpleParameterTypeBase): return self._value != other._value return self._value != other def __lt__(self,other) -> bool: if isinstance(other,_SimpleParameterTypeBase): return self._value < other._value return self._value < other def __le__(self,other) -> bool: if isinstance(other,_SimpleParameterTypeBase): return self._value <= other._value return self._value <= other def __gt__(self,other) -> bool: if isinstance(other,_SimpleParameterTypeBase): return self._value > other._value return self._value > other def __ge__(self,other) -> bool: if isinstance(other,_SimpleParameterTypeBase): return self._value >= other._value return self._value >= other class UsingBlock(_SimpleParameterTypeBase): """For injection purposes, pretend this is a new parameter type then have a post process step which strips these out """ def __init__(self,value, s:str='', loc:int=0, file:str=''): super(UsingBlock,self).__init__(value) self.s = s self.loc = loc self.file = file self.isResolved = False @staticmethod def _isValid(value) -> bool: return isinstance(value,str) def _valueFromString(value) -> str: """only used for cfg-parsing""" return str(value) def insertInto(self, parameterSet, myname:str): value = self.value() # doesn't seem to handle \0 correctly #if value == '\0': # value = '' parameterSet.addString(self.isTracked(), myname, value) def dumpPython(self, options:PrintOptions=PrintOptions()) -> str: if options.isCfg: return "process."+self.value() else: return self.value() class _Parameterizable(object): """Base class for classes which allow addition of _ParameterTypeBase data""" def __init__(self,*arg,**kargs): self.__dict__['_Parameterizable__parameterNames'] = [] self.__dict__["_isFrozen"] = False self.__dict__['_Parameterizable__validator'] = None """The named arguments are the 'parameters' which are added as 'python attributes' to the object""" if len(arg) != 0: self.__setParametersFromArg(*arg) self.__setParameters(kargs) self._isModified = False def parameterNames_(self): """Returns the name of the parameters""" return self.__parameterNames[:] def isModified(self) -> bool: if self._isModified: return True for name in self.parameterNames_(): param = self.__dict__[name] if isinstance(param, _Parameterizable) and param.isModified(): self._isModified = True return True return False def hasParameter(self, params) -> bool: """ _hasParameter_ check that pset provided has the attribute chain specified. Eg, if params is [ 'attr1', 'attr2', 'attr3' ] check for pset.attr1.attr2.attr3 returns True if parameter exists, False if not """ return (self.getParameter(params) != None) def getParameter(self, params): """ _getParameter_ Retrieve the specified parameter from the PSet Provided given the attribute chain returns None if not found """ lastParam = self # Don't accidentally iterate over letters in a string if type(params).__name__ == 'str': return getattr(self, params, None) for param in params: lastParam = getattr(lastParam, param, None) if lastParam == None: return None return lastParam def parameters_(self): """Returns a dictionary of copies of the user-set parameters""" import copy result = dict() for name in self.parameterNames_(): result[name]=copy.deepcopy(self.__dict__[name]) return result def __addParameter(self, name:str, value): if name == 'allowAnyLabel_': self.__validator = value self._isModified = True return if not isinstance(value,_ParameterTypeBase): if self.__validator is not None: value = self.__validator.convert_(value) else: self.__raiseBadSetAttr(name) if name in self.__dict__: message = "Duplicate insert of member " + name message += "\nThe original parameters are:\n" message += self.dumpPython() + '\n' raise ValueError(message) self.__dict__[name]=value self.__parameterNames.append(name) self._isModified = True def __setParametersFromArg(self, *arg): for block in arg: # Allow __PSet for testing if type(block).__name__ not in ["PSet", "__PSet", "dict"]: raise ValueError("Only PSets can be passed as unnamed argument blocks. This is a "+type(block).__name__) if isinstance(block,dict): self.__setParameters(block) else: self.__setParameters(block.parameters_()) def __setParameters(self,parameters): v = None for name,value in parameters.items(): if name == 'allowAnyLabel_': v = value continue self.__addParameter(name, value) if v is not None: self.__validator=v def hasNoParameters(self) -> bool: return len(self.__parameterNames) == 0 def __setattr__(self,name:str,value): #since labels are not supposed to have underscores at the beginning # I will assume that if we have such then we are setting an internal variable if self.isFrozen() and not (name in ["_Labelable__label","_isFrozen"] or name.startswith('_')): message = "Object already added to a process. It is read only now\n" message += " %s = %s" %(name, value) message += "\nThe original parameters are:\n" message += self.dumpPython() + '\n' raise ValueError(message) # underscored names bypass checking for _ParameterTypeBase if name[0]=='_': super(_Parameterizable,self).__setattr__(name,value) elif not name in self.__dict__: self.__addParameter(name, value) self._isModified = True else: # handle the case where users just replace with a value, a = 12, rather than a = cms.int32(12) if isinstance(value,_ParameterTypeBase): self.__dict__[name] = self.__dict__[name]._checkAndReturnValueWithType(value) else: self.__dict__[name].setValue(value) self._isModified = True def update_(self, d): """"Takes a PSet or dict and adds the entries as parameters. Already existing parameters will be overwritten. """ if type(d).__name__ not in ["PSet", "__PSet", "dict"]: raise ValueError("Only PSets or dicts can be passed to update_. This is a "+type(d).__name__) items = d.items() if isinstance(d, dict) else d.parameters_().items() for k,v in items: if v is None: delattr(self, k) else: setattr(self, k, v) def isFrozen(self) -> bool: return self._isFrozen def setIsFrozen(self): self._isFrozen = True for name in self.parameterNames_(): self.__dict__[name].setIsFrozen() def __delattr__(self,name:str): if self.isFrozen(): raise ValueError("Object already added to a process. It is read only now") super(_Parameterizable,self).__delattr__(name) self.__parameterNames.remove(name) @staticmethod def __raiseBadSetAttr(name:str): raise TypeError(name+" does not already exist, so it can only be set to a CMS python configuration type") def dumpPython(self, options:PrintOptions=PrintOptions()) -> str: specialImportRegistry.registerUse(self) sortedNames = sorted(self.parameterNames_()) if len(sortedNames) > 200: #Too many parameters for a python function call # The solution is to create a temporary dictionary which # is constructed by concatenating long lists (with maximum # 200 entries each) together. # This looks like # **dict( [(...,...), ...] + [...] + ... ) others = [] usings = [] for name in sortedNames: param = self.__dict__[name] # we don't want minuses in names name2 = name.replace('-','_') options.indent() #_UsingNodes don't get assigned variables if name.startswith("using_"): usings.append(options.indentation()+param.dumpPython(options)) else: others.append((name2, param.dumpPython(options))) options.unindent() resultList = ',\n'.join(usings) longOthers = options.indentation()+"**dict(\n" options.indent() longOthers += options.indentation()+"[\n" entriesInList = 0 options.indent() for n,v in others: entriesInList +=1 if entriesInList > 200: #need to start a new list options.unindent() longOthers += options.indentation()+"] +\n"+options.indentation()+"[\n" entriesInList = 0 options.indent() longOthers += options.indentation()+'("'+n+'" , '+v+' ),\n' longOthers += options.indentation()+"]\n" options.unindent() longOthers +=options.indentation()+")\n" options.unindent() ret = [] if resultList: ret.append(resultList) if longOthers: ret.append(longOthers) return ",\n".join(ret) #Standard case, small number of parameters others = [] usings = [] for name in sortedNames: param = self.__dict__[name] # we don't want minuses in names name2 = name.replace('-','_') options.indent() #_UsingNodes don't get assigned variables if name.startswith("using_"): usings.append(options.indentation()+param.dumpPython(options)) else: others.append(options.indentation()+name2+' = '+param.dumpPython(options)) options.unindent() # usings need to go first resultList = usings resultList.extend(others) if self.__validator is not None: options.indent() resultList.append(options.indentation()+"allowAnyLabel_="+self.__validator.dumpPython(options)) options.unindent() return ',\n'.join(resultList)+'\n' def __repr__(self) -> str: return self.dumpPython() def insertContentsInto(self, parameterSet): for name in self.parameterNames_(): param = getattr(self,name) param.insertInto(parameterSet, name) class _TypedParameterizable(_Parameterizable): """Base class for classes which are Parameterizable and have a 'type' assigned""" def __init__(self,type_,*arg,**kargs): self.__dict__['_TypedParameterizable__type'] = type_ #the 'type' is also placed in the 'arg' list and we need to remove it #if 'type_' not in kargs: # arg = arg[1:] #else: # del args['type_'] super(_TypedParameterizable,self).__init__(*arg,**kargs) saveOrigin(self, 1) def _place(self,name:str,proc): self._placeImpl(name,proc) def type_(self): """returns the type of the object, e.g. 'FooProducer'""" return self.__type def copy(self): returnValue =_TypedParameterizable.__new__(type(self)) params = self.parameters_() returnValue.__init__(self.__type,**params) returnValue._isModified = self._isModified return returnValue def clone(self, *args, **params): """Copies the object and allows one to modify the parameters of the clone. New parameters may be added by specify the exact type Modifying existing parameters can be done by just specifying the new value without having to specify the type. A parameter may be removed from the clone using the value None. #remove the parameter foo.fred mod.toModify(foo, fred = None) A parameter embedded within a PSet may be changed via a dictionary #change foo.fred.pebbles to 3 and foo.fred.friend to "barney" mod.toModify(foo, fred = dict(pebbles = 3, friend = "barney)) ) """ returnValue =_TypedParameterizable.__new__(type(self)) myparams = self.parameters_() # Prefer parameters given in PSet blocks over those in clone-from module for block in args: # Allow __PSet for testing if type(block).__name__ not in ["PSet", "__PSet"]: raise ValueError("Only PSets can be passed as unnamed argument blocks. This is a "+type(block).__name__) for name in block.parameterNames_(): try: del myparams[name] except KeyError: pass _modifyParametersFromDict(myparams, params, self._Parameterizable__raiseBadSetAttr) if self._Parameterizable__validator is not None: myparams["allowAnyLabel_"] = self._Parameterizable__validator returnValue.__init__(self.__type,*args, **myparams) returnValue._isModified = False returnValue._isFrozen = False saveOrigin(returnValue, 1) return returnValue @staticmethod def __findDefaultsFor(label:str,type): #This routine is no longer used, but I might revive it in the future import sys import glob choices = list() for d in sys.path: choices.extend(glob.glob(d+'/*/*/'+label+'.py')) if not choices: return None #now see if any of them have what we want #the use of __import__ is taken from an example # from the www.python.org documentation on __import__ for c in choices: #print " found file "+c name='.'.join(c[:-3].split('/')[-3:]) #name = c[:-3].replace('/','.') mod = __import__(name) components = name.split('.') for comp in components[1:]: mod = getattr(mod,comp) if hasattr(mod,label): default = getattr(mod,label) if isinstance(default,_TypedParameterizable): if(default.type_() == type): params = dict() for name in default.parameterNames_(): params[name] = getattr(default,name) return params return None def directDependencies(self): return [] def dumpConfig(self, options:PrintOptions=PrintOptions()) -> str: config = self.__type +' { \n' for name in self.parameterNames_(): param = self.__dict__[name] options.indent() config+=options.indentation()+param.configTypeName()+' '+name+' = '+param.configValue(options)+'\n' options.unindent() config += options.indentation()+'}\n' return config def dumpPython(self, options:PrintOptions=PrintOptions()) -> str: specialImportRegistry.registerUse(self) result = "cms."+str(type(self).__name__)+'("'+self.type_()+'"' nparam = len(self.parameterNames_()) if nparam == 0: result += ")\n" else: result += ",\n"+_Parameterizable.dumpPython(self,options)+options.indentation() + ")\n" return result def dumpPythonAttributes(self, myname:str, options:PrintOptions) -> str: """ dumps the object with all attributes declared after the constructor""" result = "" for name in sorted(self.parameterNames_()): param = self.__dict__[name] result += options.indentation() + myname + "." + name + " = " + param.dumpPython(options) + "\n" return result def nameInProcessDesc_(self, myname:str): return myname; def moduleLabel_(self, myname:str): return myname def appendToProcessDescList_(self, lst, myname:str): lst.append(self.nameInProcessDesc_(myname)) def insertInto(self, parameterSet, myname:str): newpset = parameterSet.newPSet() newpset.addString(True, "@module_label", self.moduleLabel_(myname)) newpset.addString(True, "@module_type", self.type_()) newpset.addString(True, "@module_edm_type", type(self).__name__) self.insertContentsInto(newpset) parameterSet.addPSet(True, self.nameInProcessDesc_(myname), newpset) class _Labelable(object): """A 'mixin' used to denote that the class can be paired with a label (e.g. an EDProducer)""" def label_(self) -> str: if not hasattr(self, "_Labelable__label"): if hasattr(self, '_filename'): raise RuntimeError("module has no label. Perhaps it wasn't inserted into the process?\n The module was defined at "+self._filename+":"+str(self._lineNumber)) else: raise RuntimeError("module has no label. Perhaps it wasn't inserted into the process?") return self.__label def hasLabel_(self) -> bool: return hasattr(self, "_Labelable__label") and self.__label is not None def setLabel(self,label:str): if self.hasLabel_() : if self.label_() != label and label is not None : msg100 = "Attempting to change the label of a Labelable object, possibly an attribute of the Process\n" msg101 = "Old label = "+self.label_()+" New label = "+label+"\n" msg102 = "Type = "+str(type(self))+"\n" msg103 = "Some possible solutions:\n" msg104 = " 1. Clone modules instead of using simple assignment. Cloning is\n" msg105 = " also preferred for other types when possible.\n" msg106 = " 2. Declare new names starting with an underscore if they are\n" msg107 = " for temporaries you do not want propagated into the Process. The\n" msg108 = " underscore tells \"from x import *\" and process.load not to import\n" msg109 = " the name.\n" msg110 = " 3. Reorganize so the assigment is not necessary. Giving a second\n" msg111 = " name to the same object usually causes confusion and problems.\n" msg112 = " 4. Compose Sequences: newName = cms.Sequence(oldName)\n" raise ValueError(msg100+msg101+msg102+msg103+msg104+msg105+msg106+msg107+msg108+msg109+msg110+msg111+msg112) self.__label = label def label(self) -> str: #print "WARNING: _Labelable::label() needs to be changed to label_()" return self.__label def __str__(self): #this is probably a bad idea # I added this so that when we ask a path to print # we will see the label that has been assigned return str(self.__label) def dumpSequenceConfig(self): return str(self.__label) def dumpSequencePython(self, options:PrintOptions=PrintOptions()): if options.isCfg: return 'process.'+str(self.__label) else: return str(self.__label) def _findDependencies(self,knownDeps,presentDeps): #print 'in labelled' myDeps=knownDeps.get(self.label_(),None) if myDeps!=None: if presentDeps != myDeps: raise RuntimeError("the module "+self.label_()+" has two dependencies \n" +str(presentDeps)+"\n" +str(myDeps)+"\n" +"Please modify sequences to rectify this inconsistency") else: myDeps=set(presentDeps) knownDeps[self.label_()]=myDeps presentDeps.add(self.label_()) class _Unlabelable(object): """A 'mixin' used to denote that the class can be used without a label (e.g. a Service)""" pass class _ValidatingListBase(list): """Base class for a list which enforces that its entries pass a 'validity' test""" def __init__(self,*arg,**args): super(_ValidatingListBase,self).__init__(arg) if 0 != len(args): raise SyntaxError("named arguments ("+','.join([x for x in args])+") passsed to "+str(type(self))) if not type(self)._isValid(iter(self)): raise TypeError("wrong types ("+','.join([str(type(value)) for value in iter(self)])+ ") added to "+str(type(self))) def __setitem__(self,key,value): if isinstance(key,slice): if not self._isValid(value): raise TypeError("wrong type being inserted into this container "+self._labelIfAny()) else: if not self._itemIsValid(value): raise TypeError("can not insert the type "+str(type(value))+" in container "+self._labelIfAny()) super(_ValidatingListBase,self).__setitem__(key,value) @classmethod def _isValid(cls,seq) -> bool: # see if strings get reinterpreted as lists if isinstance(seq, str): return False for item in seq: if not cls._itemIsValid(item): return False return True def _itemFromArgument(self, x): return x def _convertArguments(self, seq): if isinstance(seq, str): yield seq for x in seq: yield self._itemFromArgument(x) def append(self,x): if not self._itemIsValid(x): raise TypeError("wrong type being appended to container "+self._labelIfAny()) super(_ValidatingListBase,self).append(self._itemFromArgument(x)) def extend(self,x): if not self._isValid(x): raise TypeError("wrong type being extended to container "+self._labelIfAny()) super(_ValidatingListBase,self).extend(self._convertArguments(x)) def __add__(self,rhs): if not self._isValid(rhs): raise TypeError("wrong type being added to container "+self._labelIfAny()) import copy value = copy.copy(self) value.extend(rhs) return value def insert(self,i,x): if not self._itemIsValid(x): raise TypeError("wrong type being inserted to container "+self._labelIfAny()) super(_ValidatingListBase,self).insert(i,self._itemFromArgument(x)) def _labelIfAny(self) -> str: result = type(self).__name__ if hasattr(self, '__label'): result += ' ' + self.__label return result class _ValidatingParameterListBase(_ValidatingListBase,_ParameterTypeBase): def __init__(self,*arg,**args): _ParameterTypeBase.__init__(self) if len (arg) == 1 and not isinstance(arg[0],str): try: arg = iter(arg[0]) except TypeError: pass super(_ValidatingParameterListBase,self).__init__(*arg,**args) def value(self): return list(self) def setValue(self,v): self[:] = [] self.extend(v) self._isModified=True def configValue(self, options:PrintOptions=PrintOptions()) -> str: config = '{\n' first = True for value in iter(self): options.indent() config += options.indentation() if not first: config+=', ' config+= self.configValueForItem(value, options)+'\n' first = False options.unindent() config += options.indentation()+'}\n' return config def configValueForItem(self,item, options:PrintOptions) -> str: return str(item) def pythonValueForItem(self,item, options:PrintOptions) -> str: return self.configValueForItem(item, options) def __repr__(self): return self.dumpPython() def dumpPython(self, options:PrintOptions=PrintOptions()) -> str: specialImportRegistry.registerUse(self) result = self.pythonTypeName()+"(" n = len(self) if hasattr(self, "_nPerLine"): nPerLine = self._nPerLine else: nPerLine = 5 if n>nPerLine: options.indent() if n>=256: #wrap in a tuple since they don't have a size constraint result+=" (" wroteAtLeastOne = False for i, v in enumerate(self): wroteAtLeastOne = True if i == 0: if n>nPerLine: result += '\n'+options.indentation() else: if i % nPerLine == 0: result += ',\n'+options.indentation() else: result += ', ' result += self.pythonValueForItem(v,options) if n>=256: result +=' ) ' moreArgs = self._additionalInitArguments(options) if moreArgs: if wroteAtLeastOne: result += ', \n' + options.indentation() result += moreArgs if n>nPerLine: options.unindent() result += '\n'+options.indentation() result += ')' return result def _additionalInitArguments(self, options): return '' def directDependencies(self): return [] @staticmethod def _itemsFromStrings(strings,converter): return (converter(x).value() for x in strings) def saveOrigin(obj, level): frame = sys._getframe(level+1) obj._filename = frame.f_code.co_filename obj._lineNumber = frame.f_lineno def _modifyParametersFromDict(params, newParams, errorRaiser, keyDepth=""): if len(newParams): #need to treat items both in params and myparams specially for key,value in newParams.items(): if key in params: if value is None: del params[key] elif isinstance(value, dict): if isinstance(params[key],_Parameterizable): pset = params[key] p =pset.parameters_() oldkeys = set(p.keys()) _modifyParametersFromDict(p, value,errorRaiser, ("%s.%s" if isinstance(key, str) else "%s[%s]")%(keyDepth,key)) for k,v in p.items(): setattr(pset,k,v) oldkeys.discard(k) for k in oldkeys: delattr(pset,k) elif isinstance(params[key],_ValidatingParameterListBase): if any(not isinstance(k, int) for k in value.keys()): raise TypeError("Attempted to change a list using a dict whose keys are not integers") plist = params[key] if any((k < 0 or k >= len(plist)) for k in value.keys()): raise IndexError("Attempted to set an index which is not in the list") p = dict(enumerate(plist)) _modifyParametersFromDict(p, value,errorRaiser, ("%s.%s" if isinstance(key, str) else "%s[%s]")%(keyDepth,key)) for k,v in p.items(): plist[k] = v else: raise ValueError("Attempted to change non PSet value "+keyDepth+" using a dictionary") elif isinstance(value,_ParameterTypeBase) or (isinstance(key, int)) or isinstance(value, _Parameterizable): params[key] = value else: params[key].setValue(value) else: if isinstance(value,_ParameterTypeBase) or isinstance(value, _Parameterizable): params[key]=value else: errorRaiser(key) if __name__ == "__main__": import unittest class TestList(_ValidatingParameterListBase): @classmethod def _itemIsValid(cls,item): return True class testMixins(unittest.TestCase): def testListConstruction(self): t = TestList(1) self.assertEqual(t,[1]) t = TestList((1,)) self.assertEqual(t,[1]) t = TestList("one") self.assertEqual(t,["one"]) t = TestList( [1,]) self.assertEqual(t,[1]) t = TestList( (x for x in [1]) ) self.assertEqual(t,[1]) t = TestList(1,2) self.assertEqual(t,[1,2]) t = TestList((1,2)) self.assertEqual(t,[1,2]) t = TestList("one","two") self.assertEqual(t,["one","two"]) t = TestList(("one","two")) self.assertEqual(t,["one","two"]) t = TestList( [1,2]) self.assertEqual(t,[1,2]) t = TestList( (x for x in [1,2]) ) self.assertEqual(t,[1,2]) t = TestList( iter((1,2)) ) self.assertEqual(t,[1,2]) def testLargeList(self): #lists larger than 255 entries can not be initialized #using the constructor args = [i for i in range(0,300)] t = TestList(*args) pdump= t.dumpPython() class cms(object): def __init__(self): self.TestList = TestList pythonized = eval( pdump, globals(),{'cms':cms()} ) self.assertEqual(t,pythonized) def testUsingBlock(self): a = UsingBlock("a") self.assertTrue(isinstance(a, _ParameterTypeBase)) def testConstruction(self): class __Test(_TypedParameterizable): pass class __TestType(_SimpleParameterTypeBase): def _isValid(self,value): return True class __PSet(_ParameterTypeBase,_Parameterizable): def __init__(self,*arg,**args): #need to call the inits separately _ParameterTypeBase.__init__(self) _Parameterizable.__init__(self,*arg,**args) a = __Test("MyType", __PSet(a=__TestType(1))) self.assertEqual(a.a.value(), 1) b = __Test("MyType", __PSet(a=__TestType(1)), __PSet(b=__TestType(2))) self.assertEqual(b.a.value(), 1) self.assertEqual(b.b.value(), 2) self.assertRaises(ValueError, lambda: __Test("MyType", __PSet(a=__TestType(1)), __PSet(a=__TestType(2)))) c = __Test("MyType", dict(a=__TestType(1)), dict(b=__TestType(2))) self.assertEqual(c.a.value(), 1) self.assertEqual(c.b.value(), 2) self.assertRaises(ValueError, lambda: __Test("MyType", dict(a=__TestType(1)), dict(a=__TestType(2)))) def testUpdate_(self): class __Test(_TypedParameterizable): pass class __TestType(_SimpleParameterTypeBase): def _isValid(self,value): return True class __PSet(_ParameterTypeBase,_Parameterizable): def __init__(self,*arg,**args): #need to call the inits separately _ParameterTypeBase.__init__(self) _Parameterizable.__init__(self,*arg,**args) a = __Test("MyType", a = __TestType(1)) a.update_(dict(b=__TestType(2))) self.assertEqual(a.a.value(), 1) self.assertEqual(a.b.value(), 2) a.update_(dict(a=3)) self.assertEqual(a.a.value(), 3) a.update_(__PSet(a=__TestType(5))) self.assertEqual(a.a.value(), 5) self.assertRaises(TypeError, lambda: a.update_(dict(c=6))) self.assertTrue(hasattr(a, "a")) a.update_(dict(a=None)) self.assertFalse(hasattr(a, "a")) def testCopy(self): class __Test(_TypedParameterizable): pass class __TestType(_SimpleParameterTypeBase): def _isValid(self,value): return True a = __Test("MyType",t=__TestType(1), u=__TestType(2)) b = a.copy() self.assertEqual(b.t.value(),1) self.assertEqual(b.u.value(),2) c = __Test("MyType") self.assertEqual(len(c.parameterNames_()), 0) d = c.copy() self.assertEqual(len(d.parameterNames_()), 0) def testClone(self): class __Test(_TypedParameterizable): pass class __TestType(_SimpleParameterTypeBase): def _isValid(self,value): return True class __PSet(_ParameterTypeBase,_Parameterizable): def __init__(self,*arg,**args): #need to call the inits separately _ParameterTypeBase.__init__(self) _Parameterizable.__init__(self,*arg,**args) def dumpPython(self,options=PrintOptions()): return "__PSet(\n"+_Parameterizable.dumpPython(self, options)+options.indentation()+")" a = __Test("MyType", t=__TestType(1), u=__TestType(2), w = __TestType(3), x = __PSet(a = __TestType(4), b = __TestType(6), c = __PSet(gamma = __TestType(5)))) b = a.clone(t=3, v=__TestType(4), w= None, x = dict(a = 7, c = dict(gamma = 8), d = __TestType(9))) c = a.clone(x = dict(a=None, c=None)) self.assertEqual(a.t.value(),1) self.assertEqual(a.u.value(),2) self.assertEqual(b.t.value(),3) self.assertEqual(b.u.value(),2) self.assertEqual(b.v.value(),4) self.assertEqual(b.x.a.value(),7) self.assertEqual(b.x.b.value(),6) self.assertEqual(b.x.c.gamma.value(),8) self.assertEqual(b.x.d.value(),9) self.assertEqual(hasattr(b,"w"), False) self.assertEqual(hasattr(c.x,"a"), False) self.assertEqual(hasattr(c.x,"c"), False) self.assertRaises(TypeError,a.clone,**{"v":1}) d = a.clone(__PSet(k=__TestType(42))) self.assertEqual(d.t.value(), 1) self.assertEqual(d.k.value(), 42) d2 = a.clone(__PSet(t=__TestType(42))) self.assertEqual(d2.t.value(), 42) d3 = a.clone(__PSet(t=__TestType(42)), __PSet(u=__TestType(56))) self.assertEqual(d3.t.value(), 42) self.assertEqual(d3.u.value(), 56) self.assertRaises(ValueError,a.clone, __PSet(t=__TestType(42)), __PSet(t=__TestType(56))) d4 = a.clone(__PSet(t=__TestType(43)), u = 57) self.assertEqual(d4.t.value(), 43) self.assertEqual(d4.u.value(), 57) self.assertRaises(TypeError,a.clone,t=__TestType(43),**{"doesNotExist":57}) e = __Test("MyType") self.assertEqual(len(e.parameterNames_()), 0) f = e.clone(__PSet(a = __TestType(1)), b = __TestType(2)) self.assertEqual(f.a.value(), 1) self.assertEqual(f.b.value(), 2) g = e.clone() self.assertEqual(len(g.parameterNames_()), 0) def testModified(self): class __TestType(_SimpleParameterTypeBase): def _isValid(self,value): return True a = __TestType(1) self.assertEqual(a.isModified(),False) a.setValue(1) self.assertEqual(a.isModified(),False) a.setValue(2) self.assertEqual(a.isModified(),True) a.resetModified() self.assertEqual(a.isModified(),False) def testLargeParameterizable(self): class tLPTest(_TypedParameterizable): pass class tLPTestType(_SimpleParameterTypeBase): def _isValid(self,value): return True class __DummyModule(object): def __init__(self): self.tLPTest = tLPTest self.tLPTestType = tLPTestType p = tLPTest("MyType",** dict( [ ("a"+str(x), tLPTestType(x)) for x in range(0,300) ] ) ) #check they are the same self.assertEqual(p.dumpPython(), eval(p.dumpPython(),{"cms": __DummyModule()}).dumpPython()) def testSpecialImportRegistry(self): reg = _SpecialImportRegistry() reg.registerSpecialImportForType(int, "import foo") self.assertRaises(RuntimeError, lambda: reg.registerSpecialImportForType(int, "import bar")) reg.registerSpecialImportForType(str, "import bar") self.assertEqual(reg.getSpecialImports(), []) reg.registerUse([1]) self.assertEqual(reg.getSpecialImports(), []) reg.registerUse(1) self.assertEqual(reg.getSpecialImports(), ["import foo"]) reg.registerUse(1) self.assertEqual(reg.getSpecialImports(), ["import foo"]) reg.registerUse("a") self.assertEqual(reg.getSpecialImports(), ["import bar", "import foo"]) def testInvalidTypeChange(self): class __Test(_TypedParameterizable): pass class __TestTypeA(_SimpleParameterTypeBase): def _isValid(self,value): return True class __TestTypeB(_SimpleParameterTypeBase): def _isValid(self,value): return True pass a = __Test("MyType", t=__TestTypeA(1)) self.assertRaises(TypeError, lambda : setattr(a,'t',__TestTypeB(2))) unittest.main()