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