/usr/lib/python2.7/dist-packages/traits/tests/test_rich_compare.py is in python-traits 4.6.0-1.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
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 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 | #------------------------------------------------------------------------------
#
# Copyright (c) 2007, Enthought, Inc.
# 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!
#
#------------------------------------------------------------------------------
from __future__ import absolute_import
from traits.testing.unittest_tools import unittest
from ..api import HasTraits, Any, Str
class IdentityCompare(HasTraits):
bar = Any(rich_compare=False)
class RichCompare(HasTraits):
bar = Any(rich_compare=True)
class RichCompareTests:
def bar_changed(self, object, trait, old, new):
self.changed_object = object
self.changed_trait = trait
self.changed_old = old
self.changed_new = new
self.changed_count += 1
def reset_change_tracker(self):
self.changed_object = None
self.changed_trait = None
self.changed_old = None
self.changed_new = None
self.changed_count = 0
def check_tracker(self, object, trait, old, new, count):
self.assertEqual(count, self.changed_count)
self.assertIs(object, self.changed_object)
self.assertEqual(trait, self.changed_trait)
self.assertIs(old, self.changed_old)
self.assertIs(new, self.changed_new)
return
def test_id_first_assignment(self):
ic = IdentityCompare()
ic.on_trait_change(self.bar_changed, 'bar')
self.reset_change_tracker()
default_value = ic.bar
ic.bar = self.a
self.check_tracker(ic, 'bar', default_value, self.a, 1)
return
def test_rich_first_assignment(self):
rich = RichCompare()
rich.on_trait_change(self.bar_changed, 'bar')
self.reset_change_tracker()
default_value = rich.bar
rich.bar = self.a
self.check_tracker(rich, 'bar', default_value, self.a, 1)
return
def test_id_same_object(self):
ic = IdentityCompare()
ic.on_trait_change(self.bar_changed, 'bar')
self.reset_change_tracker()
default_value = ic.bar
ic.bar = self.a
self.check_tracker(ic, 'bar', default_value, self.a, 1)
ic.bar = self.a
self.check_tracker(ic, 'bar', default_value, self.a, 1)
return
def test_rich_same_object(self):
rich = RichCompare()
rich.on_trait_change(self.bar_changed, 'bar')
self.reset_change_tracker()
default_value = rich.bar
rich.bar = self.a
self.check_tracker(rich, 'bar', default_value, self.a, 1)
rich.bar = self.a
self.check_tracker(rich, 'bar', default_value, self.a, 1)
return
def test_id_different_object(self):
ic = IdentityCompare()
ic.on_trait_change(self.bar_changed, 'bar')
self.reset_change_tracker()
default_value = ic.bar
ic.bar = self.a
self.check_tracker(ic, 'bar', default_value, self.a, 1)
ic.bar = self.different_from_a
self.check_tracker(ic, 'bar', self.a, self.different_from_a, 2)
return
def test_rich_different_object(self):
rich = RichCompare()
rich.on_trait_change(self.bar_changed, 'bar')
self.reset_change_tracker()
default_value = rich.bar
rich.bar = self.a
self.check_tracker(rich, 'bar', default_value, self.a, 1)
rich.bar = self.different_from_a
self.check_tracker(rich, 'bar', self.a, self.different_from_a, 2)
return
def test_id_different_object_same_as(self):
ic = IdentityCompare()
ic.on_trait_change(self.bar_changed, 'bar')
self.reset_change_tracker()
default_value = ic.bar
ic.bar = self.a
self.check_tracker(ic, 'bar', default_value, self.a, 1)
ic.bar = self.same_as_a
self.check_tracker(ic, 'bar', self.a, self.same_as_a, 2)
return
def test_rich_different_object_same_as(self):
rich = RichCompare()
rich.on_trait_change(self.bar_changed, 'bar')
self.reset_change_tracker()
default_value = rich.bar
rich.bar = self.a
self.check_tracker(rich, 'bar', default_value, self.a, 1)
# Values of a and same_as_a are the same and should therefore not
# be considered a change.
rich.bar = self.same_as_a
self.check_tracker(rich, 'bar', default_value, self.a, 1)
return
class Foo(HasTraits):
name = Str
def __ne__(self, other):
# Traits uses != to do the rich compare. The default implementation
# of __ne__ is to compare the object identities.
return self.name != other.name
def __eq__(self, other):
# Not required, but a good idea to make __eq__ and __ne__ compatible
return self.name == other.name
class RichCompareHasTraitsTestCase(unittest.TestCase, RichCompareTests):
def setUp(self):
self.a = Foo(name='a')
self.same_as_a = Foo(name='a')
self.different_from_a = Foo(name='not a')
return
def test_assumptions(self):
self.assertIsNot(self.a, self.same_as_a)
self.assertIsNot(self.a, self.different_from_a)
self.assertEqual(self.a.name, self.same_as_a.name)
self.assertNotEqual(self.a.name, self.different_from_a.name)
return
### EOF
|