File: test_python_properties.py

package info (click to toggle)
python-traits 6.4.3-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 8,648 kB
  • sloc: python: 34,801; ansic: 4,266; makefile: 102
file content (56 lines) | stat: -rw-r--r-- 1,463 bytes parent folder | download
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
# (C) Copyright 2005-2023 Enthought, Inc., Austin, TX
# All rights reserved.
#
# This software is provided without warranty under the terms of the BSD
# license included in LICENSE.txt and may be redistributed only under
# the conditions described in the aforementioned license. The license
# is also available online at http://www.enthought.com/licenses/BSD.txt
#
# Thanks for using Enthought open source!

""" Tests for regular Python properties in a HasTraits subclass. """

import unittest

from traits.api import HasTraits


class Model(HasTraits):
    def __init__(self):
        super().__init__()
        self._value = 0

    @property
    def read_only(self):
        return 1729

    @property
    def value(self):
        return self._value

    @value.setter
    def value(self, new):
        self._value = new


class TestPythonProperty(unittest.TestCase):
    def test_read_only_property(self):
        model = Model()
        self.assertEqual(model.read_only, 1729)

        with self.assertRaises(AttributeError):
            model.read_only = 2034

        with self.assertRaises(AttributeError):
            del model.read_only

    def test_read_write_property(self):
        model = Model()
        self.assertEqual(model.value, 0)
        model.value = 23
        self.assertEqual(model.value, 23)
        model.value = 77
        self.assertEqual(model.value, 77)

        with self.assertRaises(AttributeError):
            del model.value