File: minimizer_api.py

package info (click to toggle)
bornagain 23.0-4
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 103,936 kB
  • sloc: cpp: 423,131; python: 40,997; javascript: 11,167; awk: 630; sh: 318; ruby: 173; xml: 130; makefile: 51; ansic: 24
file content (59 lines) | stat: -rw-r--r-- 1,801 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
"""
Testing python specific API for Minimizer related classes.
"""
import unittest
import bornagain as ba


class MinimizerAPITest(unittest.TestCase):

    def test_ParameterAttribute(self):
        """
        Testing p.value attribute
        """
        par = ba.Parameter("par", 1)
        self.assertEqual(par.value, 1)
        par.value = 42.0
        self.assertEqual(par.value, 42)

    def test_ParametersSetIterator(self):
        """
        Testing of python iterator over defined fit parameters.
        """

        pars = ba.Parameters()
        self.assertEqual(pars.size(), 0)

        pars.add(ba.Parameter("par0", 1, ba.AttLimits.limitless()))
        pars.add(ba.Parameter("par1", 2, ba.AttLimits.limitless()))
        expected_names = ["par0", "par1"]
        for index, p in enumerate(pars):
            self.assertEqual(p.name(), expected_names[index])

    def test_ParametersAdd(self):
        """
        Testing Parameters::add method
        """

        params = ba.Parameters()
        params.add("par0", 0)
        params.add("par1", 1, min=1)
        params.add("par2", 2, max=2)
        params.add("par3", 3, min=1, max=2)
        params.add("par4", 4, vary=False)

        self.assertTrue(params["par0"].limits().isLimitless())
        self.assertTrue(params["par1"].limits().isLowerLimited())
        self.assertEqual(params["par1"].limits().min(), 1)
        self.assertTrue(params["par2"].limits().isUpperLimited())
        self.assertEqual(params["par2"].limits().max(), 2)

        self.assertTrue(params["par3"].limits().isLimited())
        self.assertEqual(params["par3"].limits().min(), 1)
        self.assertEqual(params["par3"].limits().max(), 2)

        self.assertTrue(params["par4"].limits().isFixed())


if __name__ == '__main__':
    unittest.main()