Lib/test/test_operator.py

   1 import operator
   2 import unittest
   3
   4 from test import test_support
   5
   6
   7 class OperatorTestCase(unittest.TestCase):
   8     def test_lt(self):
   9         self.failIf(operator.lt(1, 0))
  10         self.failIf(operator.lt(1, 0.0))
  11         self.failIf(operator.lt(1, 1))
  12         self.failIf(operator.lt(1, 1.0))
  13         self.failUnless(operator.lt(1, 2))
  14         self.failUnless(operator.lt(1, 2.0))
  15
  16     def test_le(self):
  17         self.failIf(operator.le(1, 0))
  18         self.failIf(operator.le(1, 0.0))
  19         self.failUnless(operator.le(1, 1))
  20         self.failUnless(operator.le(1, 1.0))
  21         self.failUnless(operator.le(1, 2))
  22         self.failUnless(operator.le(1, 2.0))
  23
  24     def test_eq(self):
  25         self.failIf(operator.eq(1, 0))
  26         self.failIf(operator.eq(1, 0.0))
  27         self.failUnless(operator.eq(1, 1))
  28         self.failUnless(operator.eq(1, 1.0))
  29         self.failIf(operator.eq(1, 2))
  30         self.failIf(operator.eq(1, 2.0))
  31
  32     def test_ne(self):
  33         self.failUnless(operator.ne(1, 0))
  34         self.failUnless(operator.ne(1, 0.0))
  35         self.failIf(operator.ne(1, 1))
  36         self.failIf(operator.ne(1, 1.0))
  37         self.failUnless(operator.ne(1, 2))
  38         self.failUnless(operator.ne(1, 2.0))
  39
  40     def test_ge(self):
  41         self.failUnless(operator.ge(1, 0))
  42         self.failUnless(operator.ge(1, 0.0))
  43         self.failUnless(operator.ge(1, 1))
  44         self.failUnless(operator.ge(1, 1.0))
  45         self.failIf(operator.ge(1, 2))
  46         self.failIf(operator.ge(1, 2.0))
  47
  48     def test_gt(self):
  49         self.failUnless(operator.gt(1, 0))
  50         self.failUnless(operator.gt(1, 0.0))
  51         self.failIf(operator.gt(1, 1))
  52         self.failIf(operator.gt(1, 1.0))
  53         self.failIf(operator.gt(1, 2))
  54         self.failIf(operator.gt(1, 2.0))
  55
  56     def test_abs(self):
  57         self.failUnless(operator.abs(-1) == 1)
  58         self.failUnless(operator.abs(1) == 1)
  59
  60     def test_add(self):
  61         self.failUnless(operator.add(3, 4) == 7)
  62
  63     def test_bitwise_and(self):
  64         self.failUnless(operator.and_(0xf, 0xa) == 0xa)
  65
  66     def test_concat(self):
  67         self.failUnless(operator.concat('py', 'thon') == 'python')
  68         self.failUnless(operator.concat([1, 2], [3, 4]) == [1, 2, 3, 4])
  69
  70     def test_countOf(self):
  71         self.failUnless(operator.countOf([1, 2, 1, 3, 1, 4], 3) == 1)
  72         self.failUnless(operator.countOf([1, 2, 1, 3, 1, 4], 5) == 0)
  73
  74     def test_delitem(self):
  75         a = [4, 3, 2, 1]
  76         self.failUnless(operator.delitem(a, 1) is None)
  77         self.assert_(a == [4, 2, 1])
  78
  79     def test_delslice(self):
  80         a = range(10)
  81         self.failUnless(operator.delslice(a, 2, 8) is None)
  82         self.assert_(a == [0, 1, 8, 9])
  83
  84     def test_div(self):
  85         self.failUnless(operator.floordiv(5, 2) == 2)
  86
  87     def test_floordiv(self):
  88         self.failUnless(operator.floordiv(5, 2) == 2)
  89
  90     def test_truediv(self):
  91         self.failUnless(operator.truediv(5, 2) == 2.5)
  92
  93     def test_getitem(self):
  94         a = range(10)
  95         self.failUnless(operator.getitem(a, 2) == 2)
  96
  97     def test_getslice(self):
  98         a = range(10)
  99         self.failUnless(operator.getslice(a, 4, 6) == [4, 5])
 100
 101     def test_indexOf(self):
 102         self.failUnless(operator.indexOf([4, 3, 2, 1], 3) == 1)
 103         self.assertRaises(ValueError, operator.indexOf, [4, 3, 2, 1], 0)
 104
 105     def test_invert(self):
 106         self.failUnless(operator.inv(4) == -5)
 107
 108     def test_isCallable(self):
 109         class C:
 110             pass
 111         def check(self, o, v):
 112             self.assert_(operator.isCallable(o) == callable(o) == v)
 113         check(self, 4, 0)
 114         check(self, operator.isCallable, 1)
 115         check(self, C, 1)
 116         check(self, C(), 0)
 117
 118     def test_isMappingType(self):
 119         self.failIf(operator.isMappingType(1))
 120         self.failIf(operator.isMappingType(operator.isMappingType))
 121         self.failUnless(operator.isMappingType(operator.__dict__))
 122         self.failUnless(operator.isMappingType({}))
 123
 124     def test_isNumberType(self):
 125         self.failUnless(operator.isNumberType(8))
 126         self.failUnless(operator.isNumberType(8j))
 127         self.failUnless(operator.isNumberType(8L))
 128         self.failUnless(operator.isNumberType(8.3))
 129         self.failIf(operator.isNumberType(dir()))
 130
 131     def test_isSequenceType(self):
 132         self.failUnless(operator.isSequenceType(dir()))
 133         self.failUnless(operator.isSequenceType(()))
 134         self.failUnless(operator.isSequenceType(xrange(10)))
 135         self.failUnless(operator.isSequenceType('yeahbuddy'))
 136         self.failIf(operator.isSequenceType(3))
 137
 138     def test_lshift(self):
 139         self.failUnless(operator.lshift(5, 1) == 10)
 140         self.failUnless(operator.lshift(5, 0) == 5)
 141         self.assertRaises(ValueError, operator.lshift, 2, -1)
 142
 143     def test_mod(self):
 144         self.failUnless(operator.mod(5, 2) == 1)
 145
 146     def test_mul(self):
 147         self.failUnless(operator.mul(5, 2) == 10)
 148
 149     def test_neg(self):
 150         self.failUnless(operator.neg(5) == -5)
 151         self.failUnless(operator.neg(-5) == 5)
 152         self.failUnless(operator.neg(0) == 0)
 153         self.failUnless(operator.neg(-0) == 0)
 154
 155     def test_bitwise_or(self):
 156         self.failUnless(operator.or_(0xa, 0x5) == 0xf)
 157
 158     def test_pos(self):
 159         self.failUnless(operator.pos(5) == 5)
 160         self.failUnless(operator.pos(-5) == -5)
 161         self.failUnless(operator.pos(0) == 0)
 162         self.failUnless(operator.pos(-0) == 0)
 163
 164     def test_pow(self):
 165         self.failUnless(operator.pow(3,5) == 3**5)
 166         self.failUnless(operator.__pow__(3,5) == 3**5)
 167         self.assertRaises(TypeError, operator.pow, 1)
 168         self.assertRaises(TypeError, operator.pow, 1, 2, 3)
 169
 170     def test_repeat(self):
 171         a = range(3)
 172         self.failUnless(operator.repeat(a, 2) == a+a)
 173         self.failUnless(operator.repeat(a, 1) == a)
 174         self.failUnless(operator.repeat(a, 0) == [])
 175         a = (1, 2, 3)
 176         self.failUnless(operator.repeat(a, 2) == a+a)
 177         self.failUnless(operator.repeat(a, 1) == a)
 178         self.failUnless(operator.repeat(a, 0) == ())
 179         a = '123'
 180         self.failUnless(operator.repeat(a, 2) == a+a)
 181         self.failUnless(operator.repeat(a, 1) == a)
 182         self.failUnless(operator.repeat(a, 0) == '')
 183
 184     def test_rshift(self):
 185         self.failUnless(operator.rshift(5, 1) == 2)
 186         self.failUnless(operator.rshift(5, 0) == 5)
 187         self.assertRaises(ValueError, operator.rshift, 2, -1)
 188
 189     def test_contains(self):
 190         self.failUnless(operator.contains(range(4), 2))
 191         self.failIf(operator.contains(range(4), 5))
 192         self.failUnless(operator.sequenceIncludes(range(4), 2))
 193         self.failIf(operator.sequenceIncludes(range(4), 5))
 194
 195     def test_setitem(self):
 196         a = range(3)
 197         self.failUnless(operator.setitem(a, 0, 2) is None)
 198         self.assert_(a == [2, 1, 2])
 199         self.assertRaises(IndexError, operator.setitem, a, 4, 2)
 200
 201     def test_setslice(self):
 202         a = range(4)
 203         self.failUnless(operator.setslice(a, 1, 3, [2, 1]) is None)
 204         self.assert_(a == [0, 2, 1, 3])
 205
 206     def test_sub(self):
 207         self.failUnless(operator.sub(5, 2) == 3)
 208
 209     def test_truth(self):
 210         self.failUnless(operator.truth(5))
 211         self.failUnless(operator.truth([0]))
 212         self.failIf(operator.truth(0))
 213         self.failIf(operator.truth([]))
 214
 215     def test_bitwise_xor(self):
 216         self.failUnless(operator.xor(0xb, 0xc) == 0x7)
 217
 218     def test_is(self):
 219         a = b = 'xyzpdq'
 220         c = a[:3] + b[3:]
 221         self.failUnless(operator.is_(a, b))
 222         self.failIf(operator.is_(a,c))
 223
 224     def test_is_not(self):
 225         a = b = 'xyzpdq'
 226         c = a[:3] + b[3:]
 227         self.failIf(operator.is_not(a, b))
 228         self.failUnless(operator.is_not(a,c))
 229
 230 def test_main():
 231     test_support.run_unittest(OperatorTestCase)
 232
 233
 234 if __name__ == "__main__":
 235     test_main()