コード例 #1
0
ファイル: core.py プロジェクト: schlamar/sitforc
    def _load(self):
        '''
        Loads the defined models from "modellib.sfm".
        '''
        folder = os.path.dirname(__file__)
        try:
            config = ConfigObj(os.path.join(folder, 'modellib.sfm'))
        except configobj.ConfigObjError as e:
            txt = 'File "modellib.sfm" has an error: {0}'.format(e)
            raise configobj.ConfigObjError(txt)
        if not config:
            warn('File "modellib.sfm" could not read or is empty. '
                 'No models in modellib.', SitforcWarning)
        for modelname in config:
            params = dict()
            comment = ''
            funcstring = ''
            for key in config[modelname]:
                if key == 'func':
                    funcstring = config[modelname][key]
                elif key == 'comment':
                    comment = config[modelname][key]
                else:
                    params[key] = config[modelname].as_float(key)
            if not funcstring:
                warn('Function not defined for model "{0}" '
                     '(in "modellib.sfm").'.format(modelname), 
                     SitforcWarning)
                continue
            try:
                func, latex, ident_params = parse_func(funcstring)
            except ParseException as e:
                warn('Function for model "{0}" in "modellib.sfm" has '
                     'an error: {1}'.format(modelname, e), 
                     SitforcWarning)
                continue

            try:
                func(1, params)
            except KeyError as e:
                warn('Param {0} for model "{1}" is not defined '
                     'in "modellib.sfm".'.format(e, modelname), 
                     SitforcWarning)
                continue
            
            self.lib[modelname] = Model(modelname, func, 
                                        funcstring, latex, 
                                        **params)
            self.lib[modelname].comment = comment
コード例 #2
0
ファイル: test_funcparser.py プロジェクト: schlamar/sitforc
    def test_parse_func(self):
        # exceptions
        
        funcstring = 'open("/etc/passwd").read()' # try injection
        self.assertRaises(ParseException, parse_func, funcstring)
        
        funcstring = '(2.3 + 3' # general syntax error
        self.assertRaises(ParseException, parse_func, funcstring)
        
        funcstring = '2 * x + 3 * y' # using not allowed keyword (1)
        self.assertRaises(ParseException, parse_func, funcstring)
        
        funcstring = '2 * 3 + t["dx"] * x' # using not allowed keyword (2)
        self.assertRaises(ParseException, parse_func, funcstring)
        
        # check resulting values
        
        funcstring = 'p["dx"]'
        f = parse_func(funcstring)[0]
        for value in [-1, 0, 1, 2, 3]:
            p = dict()
            p['dx'] = value
            self.assertEqual(f(0, p), value)
            
        funcstring = 'x'
        f = parse_func(funcstring)[0]
        g = lambda x: x
        for value in xrange(-5, 5):
            self.assertEqual(f(value, None), g(value))
            
        funcstring = '(x - 3) * (x + 2)'
        f = parse_func(funcstring)[0]
        funcstring = 'x**2 - x - 6' # funcs are equivalent
        g = parse_func(funcstring)[0]
        for value in xrange(-5, 5):
            self.assertEqual(f(value, None), g(value, None))
        self.assertEqual(f(3, None), 0) # root1
        self.assertEqual(f(-2, None), 0) # root2
        
        # check generated latex expressions
        
        funcstring = '2 + 3'
        latex = parse_func(funcstring)[1]
        self.assertEqual(latex, r'$(2 + 3)$')
        
        funcstring = '2 - 3'
        latex = parse_func(funcstring)[1]
        self.assertEqual(latex, r'$(2 - 3)$')
        
        funcstring = '2 * 3'
        latex = parse_func(funcstring)[1]
        self.assertEqual(latex, r'$2 \cdot 3$')

        funcstring = '2 / 3'
        latex = parse_func(funcstring)[1]
        self.assertEqual(latex, r'$\frac{2}{3}$')
        
        funcstring = '2 ** 3'
        latex = parse_func(funcstring)[1]
        self.assertEqual(latex, r'$(2)^{3}$')
        
        funcstring = '-(2 + 3)'
        latex = parse_func(funcstring)[1]
        self.assertEqual(latex, r'$-((2 + 3))$')
        
        funcstring = '-(x)'
        latex = parse_func(funcstring)[1]
        self.assertEqual(latex, r'$-(x)$')
        
        funcstring = 'x + p["c"]'
        latex = parse_func(funcstring)[1]
        self.assertEqual(latex, r'$(x + c)$')
        
        funcstring = 'exp(2)'
        latex = parse_func(funcstring)[1]
        self.assertEqual(latex, r'$\mathrm{e}^{2}$')
        
        funcstring = 'sin(x)'
        latex = parse_func(funcstring)[1]
        self.assertEqual(latex, r'$\sin{x}$')
        
        funcstring = 'fac(x)'
        latex = parse_func(funcstring)[1]
        self.assertEqual(latex, r'$x\mathrm{!}$')
        
        # check identified parameters
        
        params = ['dx']
        funcstring = 'p["dx"]'
        identified_params = parse_func(funcstring)[2]
        for param in params:
            self.assertTrue(param in identified_params)
            
        params = ['t1', 't2', 't3']
        funcstring = 'p["t1"] + p["t2"] + p["t3"]'
        identified_params = parse_func(funcstring)[2]
        for param in params:
            self.assertTrue(param in identified_params)