diffpy.srfit/tests/test-parameter.py at 87533915b6af6512c98a05da93398b797f2abf74 · stevenhua0320/diffpy.srfit · GitHub

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
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
#!/usr/bin/env python
##############################################################################
#
# diffpy.srfit      by DANSE Diffraction group
#                   Simon J. L. Billinge
#                   (c) 2010 The Trustees of Columbia University
#                   in the City of New York.  All rights reserved.
#
# File coded by:    Pavol Juhas
#
# See AUTHORS.txt for a list of people who contributed.
# See LICENSE_DANSE.txt for license information.
#
##############################################################################
"""Tests for refinableobj module."""

import unittest

from diffpy.srfit.fitbase.parameter import (
    Parameter,
    ParameterAdapter,
    ParameterProxy,
)


class TestParameter(unittest.TestCase):

    def testSetValue(self):
        """Test initialization."""
        par_l = Parameter("l")

        par_l.setValue(3.14)
        self.assertAlmostEqual(3.14, par_l.getValue())

        # Try array
        import numpy

        x = numpy.arange(0, 10, 0.1)
        par_l.setValue(x)
        self.assertTrue(par_l.getValue() is x)
        self.assertTrue(par_l.value is x)

        # Change the array
        y = numpy.arange(0, 10, 0.5)
        par_l.value = y
        self.assertTrue(par_l.getValue() is y)
        self.assertTrue(par_l.value is y)

        # Back to scalar
        par_l.setValue(1.01)
        self.assertAlmostEqual(1.01, par_l.getValue())
        self.assertAlmostEqual(1.01, par_l.value)
        return


class TestParameterProxy(unittest.TestCase):

    def testProxy(self):
        """Test the ParameterProxy class."""
        par_l = Parameter("l", 3.14)

        # Try Accessor adaptation
        la = ParameterProxy("l2", par_l)

        self.assertEqual("l2", la.name)
        self.assertEqual(par_l.getValue(), la.getValue())

        # Change the parameter
        par_l.value = 2.3
        self.assertEqual(par_l.getValue(), la.getValue())
        self.assertEqual(par_l.value, la.value)

        # Change the proxy
        la.value = 3.2
        self.assertEqual(par_l.getValue(), la.getValue())
        self.assertEqual(par_l.value, la.value)

        return


class TestParameterAdapter(unittest.TestCase):

    def testWrapper(self):
        """Test the adapter.

        This adapts a Parameter to the Parameter interface. :)
        """
        par_l = Parameter("l", 3.14)

        # Try Accessor adaptation
        la = ParameterAdapter(
            "l", par_l, getter=Parameter.getValue, setter=Parameter.setValue
        )

        self.assertEqual(par_l.name, la.name)
        self.assertEqual(par_l.getValue(), la.getValue())

        # Change the parameter
        par_l.setValue(2.3)
        self.assertEqual(par_l.getValue(), la.getValue())

        # Change the adapter
        la.setValue(3.2)
        self.assertEqual(par_l.getValue(), la.getValue())

        # Try Attribute adaptation
        la = ParameterAdapter("l", par_l, attr="value")

        self.assertEqual(par_l.name, la.name)
        self.assertEqual("value", la.attr)
        self.assertEqual(par_l.getValue(), la.getValue())

        # Change the parameter
        par_l.setValue(2.3)
        self.assertEqual(par_l.getValue(), la.getValue())

        # Change the adapter
        la.setValue(3.2)
        self.assertEqual(par_l.getValue(), la.getValue())

        return


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