Guido van Rossum | 1daf954 | 2007-08-30 17:45:54 +0000 | [diff] [blame] | 1 | """Unit tests for numbers.py.""" |
| 2 | |
Christian Heimes | 400adb0 | 2008-02-01 08:12:03 +0000 | [diff] [blame^] | 3 | import math |
Guido van Rossum | 1daf954 | 2007-08-30 17:45:54 +0000 | [diff] [blame] | 4 | import operator |
Christian Heimes | 400adb0 | 2008-02-01 08:12:03 +0000 | [diff] [blame^] | 5 | import unittest |
| 6 | from numbers import Complex, Real, Rational, Integral |
| 7 | from numbers import Exact, Inexact |
| 8 | from numbers import Number |
| 9 | from test import test_support |
Guido van Rossum | 1daf954 | 2007-08-30 17:45:54 +0000 | [diff] [blame] | 10 | |
| 11 | class TestNumbers(unittest.TestCase): |
| 12 | def test_int(self): |
| 13 | self.failUnless(issubclass(int, Integral)) |
| 14 | self.failUnless(issubclass(int, Complex)) |
| 15 | self.failUnless(issubclass(int, Exact)) |
| 16 | self.failIf(issubclass(int, Inexact)) |
| 17 | |
| 18 | self.assertEqual(7, int(7).real) |
| 19 | self.assertEqual(0, int(7).imag) |
| 20 | self.assertEqual(7, int(7).conjugate()) |
| 21 | self.assertEqual(7, int(7).numerator) |
| 22 | self.assertEqual(1, int(7).denominator) |
| 23 | |
| 24 | def test_float(self): |
| 25 | self.failIf(issubclass(float, Rational)) |
| 26 | self.failUnless(issubclass(float, Real)) |
| 27 | self.failIf(issubclass(float, Exact)) |
| 28 | self.failUnless(issubclass(float, Inexact)) |
| 29 | |
| 30 | self.assertEqual(7.3, float(7.3).real) |
| 31 | self.assertEqual(0, float(7.3).imag) |
| 32 | self.assertEqual(7.3, float(7.3).conjugate()) |
| 33 | |
| 34 | def test_complex(self): |
| 35 | self.failIf(issubclass(complex, Real)) |
| 36 | self.failUnless(issubclass(complex, Complex)) |
| 37 | self.failIf(issubclass(complex, Exact)) |
| 38 | self.failUnless(issubclass(complex, Inexact)) |
| 39 | |
| 40 | c1, c2 = complex(3, 2), complex(4,1) |
Christian Heimes | 400adb0 | 2008-02-01 08:12:03 +0000 | [diff] [blame^] | 41 | # XXX: This is not ideal, but see the comment in math_trunc(). |
| 42 | self.assertRaises(TypeError, math.trunc, c1) |
Guido van Rossum | 1daf954 | 2007-08-30 17:45:54 +0000 | [diff] [blame] | 43 | self.assertRaises(TypeError, operator.mod, c1, c2) |
| 44 | self.assertRaises(TypeError, divmod, c1, c2) |
| 45 | self.assertRaises(TypeError, operator.floordiv, c1, c2) |
| 46 | self.assertRaises(TypeError, float, c1) |
| 47 | self.assertRaises(TypeError, int, c1) |
| 48 | |
| 49 | def test_main(): |
| 50 | test_support.run_unittest(TestNumbers) |
| 51 | |
| 52 | |
| 53 | if __name__ == "__main__": |
| 54 | unittest.main() |