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 |
Guido van Rossum | 1daf954 | 2007-08-30 17:45:54 +0000 | [diff] [blame] | 7 | |
| 8 | class TestNumbers(unittest.TestCase): |
| 9 | def test_int(self): |
Benjamin Peterson | c9c0f20 | 2009-06-30 23:06:06 +0000 | [diff] [blame] | 10 | self.assertTrue(issubclass(int, Integral)) |
| 11 | self.assertTrue(issubclass(int, Complex)) |
Guido van Rossum | 1daf954 | 2007-08-30 17:45:54 +0000 | [diff] [blame] | 12 | |
| 13 | self.assertEqual(7, int(7).real) |
| 14 | self.assertEqual(0, int(7).imag) |
| 15 | self.assertEqual(7, int(7).conjugate()) |
Benjamin Peterson | 15ac311 | 2011-05-30 19:22:53 -0500 | [diff] [blame] | 16 | self.assertEqual(-7, int(-7).conjugate()) |
Guido van Rossum | 1daf954 | 2007-08-30 17:45:54 +0000 | [diff] [blame] | 17 | self.assertEqual(7, int(7).numerator) |
| 18 | self.assertEqual(1, int(7).denominator) |
| 19 | |
| 20 | def test_float(self): |
Benjamin Peterson | c9c0f20 | 2009-06-30 23:06:06 +0000 | [diff] [blame] | 21 | self.assertFalse(issubclass(float, Rational)) |
| 22 | self.assertTrue(issubclass(float, Real)) |
Guido van Rossum | 1daf954 | 2007-08-30 17:45:54 +0000 | [diff] [blame] | 23 | |
| 24 | self.assertEqual(7.3, float(7.3).real) |
| 25 | self.assertEqual(0, float(7.3).imag) |
| 26 | self.assertEqual(7.3, float(7.3).conjugate()) |
Benjamin Peterson | 15ac311 | 2011-05-30 19:22:53 -0500 | [diff] [blame] | 27 | self.assertEqual(-7.3, float(-7.3).conjugate()) |
Guido van Rossum | 1daf954 | 2007-08-30 17:45:54 +0000 | [diff] [blame] | 28 | |
| 29 | def test_complex(self): |
Benjamin Peterson | c9c0f20 | 2009-06-30 23:06:06 +0000 | [diff] [blame] | 30 | self.assertFalse(issubclass(complex, Real)) |
| 31 | self.assertTrue(issubclass(complex, Complex)) |
Guido van Rossum | 1daf954 | 2007-08-30 17:45:54 +0000 | [diff] [blame] | 32 | |
| 33 | c1, c2 = complex(3, 2), complex(4,1) |
Christian Heimes | 400adb0 | 2008-02-01 08:12:03 +0000 | [diff] [blame] | 34 | # XXX: This is not ideal, but see the comment in math_trunc(). |
| 35 | self.assertRaises(TypeError, math.trunc, c1) |
Guido van Rossum | 1daf954 | 2007-08-30 17:45:54 +0000 | [diff] [blame] | 36 | self.assertRaises(TypeError, operator.mod, c1, c2) |
| 37 | self.assertRaises(TypeError, divmod, c1, c2) |
| 38 | self.assertRaises(TypeError, operator.floordiv, c1, c2) |
| 39 | self.assertRaises(TypeError, float, c1) |
| 40 | self.assertRaises(TypeError, int, c1) |
| 41 | |
Guido van Rossum | 1daf954 | 2007-08-30 17:45:54 +0000 | [diff] [blame] | 42 | |
| 43 | if __name__ == "__main__": |
| 44 | unittest.main() |