def testUniqueVarsTwo(self): from iegen.util import get_unique_vars vars=['a','a'] unique_vars=[] used_vars=set(['a']) var_map={} get_unique_vars(vars,unique_vars,used_vars,var_map) self.failUnless(vars==['a','a']) self.failUnless(unique_vars==['a0','a1']) self.failUnless(used_vars==set(['a','a0','a1']))
def testUniqueVarsSome(self): from iegen.util import get_unique_vars vars=['a','b','c'] unique_vars=[] used_vars=set(['a','b','c']) var_map={} get_unique_vars(vars,unique_vars,used_vars,var_map) self.failUnless(vars==['a','b','c']) self.failUnless(unique_vars==['a0','b0','c0']) self.failUnless(used_vars==set(['a','b','c','a0','b0','c0'])) self.failUnless(var_map=={'a':'a0','b':'b0','c':'c0'})