Changeset e65c3ba in sasmodels for sasmodels/convert.py


Ignore:
Timestamp:
Nov 28, 2017 4:09:34 PM (6 years ago)
Author:
Paul Kienzle <pkienzle@…>
Branches:
master, core_shell_microgels, magnetic_model, ticket-1257-vesicle-product, ticket_1156, ticket_1265_superball, ticket_822_more_unit_tests
Children:
32398dc
Parents:
110f69c
Message:

lint

File:
1 edited

Legend:

Unmodified
Added
Removed
  • sasmodels/convert.py

    r07c8d46 re65c3ba  
    44from __future__ import print_function, division 
    55 
    6 import re 
    76import math 
    87import warnings 
     
    6463    return [pk*scale for pk in par] if isinstance(par, list) else par*scale 
    6564 
    66 def _is_sld(model_info, id): 
     65def _is_sld(model_info, par): 
    6766    """ 
    6867    Return True if parameter is a magnetic magnitude or SLD parameter. 
    6968    """ 
    70     if id.startswith('M0:'): 
     69    if par.startswith('M0:'): 
    7170        return True 
    72     if '_pd' in id or '.' in id: 
     71    if '_pd' in par or '.' in par: 
    7372        return False 
    7473    for p in model_info.parameters.call_parameters: 
    75         if p.id == id: 
     74        if p.id == par: 
    7675            return p.type == 'sld' 
    7776    # check through kernel parameters in case it is a named as a vector 
    7877    for p in model_info.parameters.kernel_parameters: 
    79         if p.id == id: 
     78        if p.id == par: 
    8079            return p.type == 'sld' 
    8180    return False 
     
    8887    *scale=1e-6*.  For forward conversion use *scale=1e6*. 
    8988    """ 
    90     return dict((id, (_rescale(v, scale) if _is_sld(model_info, id) else v)) 
    91                 for id, v in pars.items()) 
    92  
    93  
    94 def _get_translation_table(model_info, version=(3,1,2)): 
     89    return dict((par, (_rescale(v, scale) if _is_sld(model_info, par) else v)) 
     90                for par, v in pars.items()) 
     91 
     92 
     93def _get_translation_table(model_info, version=(3, 1, 2)): 
    9594    conv_param = CONVERSION_TABLE.get(version, {}).get(model_info.id, [None, {}]) 
    9695    translation = conv_param[1].copy() 
     
    130129    newpars = pars.copy() 
    131130    for new, old in mapping.items(): 
    132         if old == new: continue 
    133         if old is None: continue 
    134         for underscore, dot in PD_DOT: 
     131        if old == new: 
     132            continue 
     133        if old is None: 
     134            continue 
     135        for _, dot in PD_DOT: 
    135136            source = old+dot 
    136137            if source in newpars: 
     
    159160    return None 
    160161 
    161 def _hand_convert(name, oldpars, version=(3,1,2)): 
    162     if version == (3,1,2): 
     162def _hand_convert(name, oldpars, version=(3, 1, 2)): 
     163    if version == (3, 1, 2): 
    163164        oldpars = _hand_convert_3_1_2_to_4_1(name, oldpars) 
    164165    return oldpars 
     
    295296    return oldpars 
    296297 
    297 def convert_model(name, pars, use_underscore=False, model_version=(3,1,2)): 
     298def convert_model(name, pars, use_underscore=False, model_version=(3, 1, 2)): 
    298299    """ 
    299300    Convert model from old style parameter names to new style. 
     
    327328        newpars = _convert_pars(newpars, translation) 
    328329        # TODO: Still not convinced this is the best check 
    329         if not model_info.structure_factor and version == (3,1,2): 
     330        if not model_info.structure_factor and version == (3, 1, 2): 
    330331            newpars = _rescale_sld(model_info, newpars, 1e6) 
    331332        newpars.setdefault('scale', 1.0) 
     
    615616        print("==== %s out ====="%new_name) 
    616617        print(str(compare.parlist(model_info, new_pars, True))) 
    617     assert name==new_name, "%r != %r"%(name, new_name) 
     618    assert name == new_name, "%r != %r"%(name, new_name) 
    618619    for k, v in new_pars.items(): 
    619620        assert k in pars, "%s: %r appeared from conversion"%(name, k) 
    620621        if isinstance(v, float): 
    621             assert abs(v-pars[k])<=abs(1e-12*v), "%s: %r  %s != %s"%(name, k, v, pars[k]) 
     622            assert abs(v-pars[k]) <= abs(1e-12*v), \ 
     623                "%s: %r  %s != %s"%(name, k, v, pars[k]) 
    622624        else: 
    623625            assert v == pars[k], "%s: %r  %s != %s"%(name, k, v, pars[k]) 
Note: See TracChangeset for help on using the changeset viewer.