blob: 452f18126de9ae9f9e7ffd7d68c370da057ca6bb [file] [log] [blame]
Christian Heimes3feef612008-02-11 06:19:17 +00001"""Tests for Lib/fractions.py."""
Guido van Rossum7736b5b2008-01-15 21:44:53 +00002
3from decimal import Decimal
Zachary Ware38c707e2015-04-13 15:00:43 -05004from test.support import requires_IEEE_754
Guido van Rossum7736b5b2008-01-15 21:44:53 +00005import math
Mark Dickinson85424c92009-07-18 14:41:42 +00006import numbers
Guido van Rossum7736b5b2008-01-15 21:44:53 +00007import operator
Christian Heimes3feef612008-02-11 06:19:17 +00008import fractions
Ezio Melotti682d3742012-02-29 14:05:53 +02009import sys
Guido van Rossum7736b5b2008-01-15 21:44:53 +000010import unittest
Serhiy Storchaka48e47aa2015-05-13 00:19:51 +030011import warnings
Christian Heimes969fe572008-01-25 11:23:10 +000012from copy import copy, deepcopy
Neal Norwitz6ff93fe2008-01-26 21:52:30 +000013from pickle import dumps, loads
Christian Heimes68f5fbe2008-02-14 08:27:37 +000014F = fractions.Fraction
Christian Heimes3feef612008-02-11 06:19:17 +000015gcd = fractions.gcd
Christian Heimesaf98da12008-01-27 15:18:18 +000016
Mark Dickinson85424c92009-07-18 14:41:42 +000017class DummyFloat(object):
18 """Dummy float class for testing comparisons with Fractions"""
19
20 def __init__(self, value):
21 if not isinstance(value, float):
22 raise TypeError("DummyFloat can only be initialized from float")
23 self.value = value
24
25 def _richcmp(self, other, op):
26 if isinstance(other, numbers.Rational):
27 return op(F.from_float(self.value), other)
28 elif isinstance(other, DummyFloat):
29 return op(self.value, other.value)
30 else:
31 return NotImplemented
32
33 def __eq__(self, other): return self._richcmp(other, operator.eq)
34 def __le__(self, other): return self._richcmp(other, operator.le)
35 def __lt__(self, other): return self._richcmp(other, operator.lt)
36 def __ge__(self, other): return self._richcmp(other, operator.ge)
37 def __gt__(self, other): return self._richcmp(other, operator.gt)
38
39 # shouldn't be calling __float__ at all when doing comparisons
40 def __float__(self):
41 assert False, "__float__ should not be invoked for comparisons"
42
43 # same goes for subtraction
44 def __sub__(self, other):
45 assert False, "__sub__ should not be invoked for comparisons"
46 __rsub__ = __sub__
47
48
49class DummyRational(object):
50 """Test comparison of Fraction with a naive rational implementation."""
51
52 def __init__(self, num, den):
Serhiy Storchaka48e47aa2015-05-13 00:19:51 +030053 g = math.gcd(num, den)
Mark Dickinson85424c92009-07-18 14:41:42 +000054 self.num = num // g
55 self.den = den // g
56
57 def __eq__(self, other):
58 if isinstance(other, fractions.Fraction):
59 return (self.num == other._numerator and
60 self.den == other._denominator)
61 else:
62 return NotImplemented
63
64 def __lt__(self, other):
65 return(self.num * other._denominator < self.den * other._numerator)
66
67 def __gt__(self, other):
68 return(self.num * other._denominator > self.den * other._numerator)
69
70 def __le__(self, other):
71 return(self.num * other._denominator <= self.den * other._numerator)
72
73 def __ge__(self, other):
74 return(self.num * other._denominator >= self.den * other._numerator)
75
76 # this class is for testing comparisons; conversion to float
77 # should never be used for a comparison, since it loses accuracy
78 def __float__(self):
79 assert False, "__float__ should not be invoked"
Christian Heimesaf98da12008-01-27 15:18:18 +000080
Ezio Melotti682d3742012-02-29 14:05:53 +020081class DummyFraction(fractions.Fraction):
82 """Dummy Fraction subclass for copy and deepcopy testing."""
83
Christian Heimesaf98da12008-01-27 15:18:18 +000084class GcdTest(unittest.TestCase):
85
86 def testMisc(self):
Serhiy Storchaka48e47aa2015-05-13 00:19:51 +030087 # fractions.gcd() is deprecated
88 with self.assertWarnsRegex(DeprecationWarning, r'fractions\.gcd'):
89 gcd(1, 1)
90 with warnings.catch_warnings():
91 warnings.filterwarnings('ignore', r'fractions\.gcd',
92 DeprecationWarning)
93 self.assertEqual(0, gcd(0, 0))
94 self.assertEqual(1, gcd(1, 0))
95 self.assertEqual(-1, gcd(-1, 0))
96 self.assertEqual(1, gcd(0, 1))
97 self.assertEqual(-1, gcd(0, -1))
98 self.assertEqual(1, gcd(7, 1))
99 self.assertEqual(-1, gcd(7, -1))
100 self.assertEqual(1, gcd(-23, 15))
101 self.assertEqual(12, gcd(120, 84))
102 self.assertEqual(-12, gcd(84, -120))
103 self.assertEqual(gcd(120.0, 84), 12.0)
104 self.assertEqual(gcd(120, 84.0), 12.0)
105 self.assertEqual(gcd(F(120), F(84)), F(12))
106 self.assertEqual(gcd(F(120, 77), F(84, 55)), F(12, 385))
Christian Heimesaf98da12008-01-27 15:18:18 +0000107
Guido van Rossum7736b5b2008-01-15 21:44:53 +0000108
109def _components(r):
110 return (r.numerator, r.denominator)
111
Christian Heimesaf98da12008-01-27 15:18:18 +0000112
Christian Heimes3feef612008-02-11 06:19:17 +0000113class FractionTest(unittest.TestCase):
Guido van Rossum7736b5b2008-01-15 21:44:53 +0000114
115 def assertTypedEquals(self, expected, actual):
116 """Asserts that both the types and values are the same."""
Ezio Melottib3aedd42010-11-20 19:04:17 +0000117 self.assertEqual(type(expected), type(actual))
118 self.assertEqual(expected, actual)
Guido van Rossum7736b5b2008-01-15 21:44:53 +0000119
120 def assertRaisesMessage(self, exc_type, message,
121 callable, *args, **kwargs):
122 """Asserts that callable(*args, **kwargs) raises exc_type(message)."""
123 try:
124 callable(*args, **kwargs)
125 except exc_type as e:
Ezio Melottib3aedd42010-11-20 19:04:17 +0000126 self.assertEqual(message, str(e))
Guido van Rossum7736b5b2008-01-15 21:44:53 +0000127 else:
128 self.fail("%s not raised" % exc_type.__name__)
129
130 def testInit(self):
Ezio Melottib3aedd42010-11-20 19:04:17 +0000131 self.assertEqual((0, 1), _components(F()))
132 self.assertEqual((7, 1), _components(F(7)))
133 self.assertEqual((7, 3), _components(F(F(7, 3))))
Guido van Rossum7736b5b2008-01-15 21:44:53 +0000134
Ezio Melottib3aedd42010-11-20 19:04:17 +0000135 self.assertEqual((-1, 1), _components(F(-1, 1)))
136 self.assertEqual((-1, 1), _components(F(1, -1)))
137 self.assertEqual((1, 1), _components(F(-2, -2)))
138 self.assertEqual((1, 2), _components(F(5, 10)))
139 self.assertEqual((7, 15), _components(F(7, 15)))
140 self.assertEqual((10**23, 1), _components(F(10**23)))
Guido van Rossum7736b5b2008-01-15 21:44:53 +0000141
Ezio Melottib3aedd42010-11-20 19:04:17 +0000142 self.assertEqual((3, 77), _components(F(F(3, 7), 11)))
143 self.assertEqual((-9, 5), _components(F(2, F(-10, 9))))
144 self.assertEqual((2486, 2485), _components(F(F(22, 7), F(355, 113))))
Mark Dickinsond4d95f82009-04-24 14:06:19 +0000145
Christian Heimes3feef612008-02-11 06:19:17 +0000146 self.assertRaisesMessage(ZeroDivisionError, "Fraction(12, 0)",
Christian Heimes68f5fbe2008-02-14 08:27:37 +0000147 F, 12, 0)
Georg Brandl86b2fb92008-07-16 03:43:04 +0000148 self.assertRaises(TypeError, F, 1.5 + 3j)
Guido van Rossum7736b5b2008-01-15 21:44:53 +0000149
Georg Brandl86b2fb92008-07-16 03:43:04 +0000150 self.assertRaises(TypeError, F, "3/2", 3)
Mark Dickinsond4d95f82009-04-24 14:06:19 +0000151 self.assertRaises(TypeError, F, 3, 0j)
152 self.assertRaises(TypeError, F, 3, 1j)
Mark Dickinson7caf9082016-08-23 16:16:52 +0100153 self.assertRaises(TypeError, F, 1, 2, 3)
Mark Dickinsond4d95f82009-04-24 14:06:19 +0000154
Mark Dickinson98127c32010-04-03 11:18:52 +0000155 @requires_IEEE_754
156 def testInitFromFloat(self):
Ezio Melottib3aedd42010-11-20 19:04:17 +0000157 self.assertEqual((5, 2), _components(F(2.5)))
158 self.assertEqual((0, 1), _components(F(-0.0)))
159 self.assertEqual((3602879701896397, 36028797018963968),
160 _components(F(0.1)))
Mark Dickinson73726aa2012-11-15 20:58:40 +0000161 # bug 16469: error types should be consistent with float -> int
162 self.assertRaises(ValueError, F, float('nan'))
163 self.assertRaises(OverflowError, F, float('inf'))
164 self.assertRaises(OverflowError, F, float('-inf'))
Mark Dickinson98127c32010-04-03 11:18:52 +0000165
166 def testInitFromDecimal(self):
Ezio Melottib3aedd42010-11-20 19:04:17 +0000167 self.assertEqual((11, 10),
168 _components(F(Decimal('1.1'))))
169 self.assertEqual((7, 200),
170 _components(F(Decimal('3.5e-2'))))
171 self.assertEqual((0, 1),
172 _components(F(Decimal('.000e20'))))
Mark Dickinson73726aa2012-11-15 20:58:40 +0000173 # bug 16469: error types should be consistent with decimal -> int
174 self.assertRaises(ValueError, F, Decimal('nan'))
175 self.assertRaises(ValueError, F, Decimal('snan'))
176 self.assertRaises(OverflowError, F, Decimal('inf'))
177 self.assertRaises(OverflowError, F, Decimal('-inf'))
Christian Heimes587c2bf2008-01-19 16:21:02 +0000178
179 def testFromString(self):
Ezio Melottib3aedd42010-11-20 19:04:17 +0000180 self.assertEqual((5, 1), _components(F("5")))
181 self.assertEqual((3, 2), _components(F("3/2")))
182 self.assertEqual((3, 2), _components(F(" \n +3/2")))
183 self.assertEqual((-3, 2), _components(F("-3/2 ")))
184 self.assertEqual((13, 2), _components(F(" 013/02 \n ")))
185 self.assertEqual((16, 5), _components(F(" 3.2 ")))
186 self.assertEqual((-16, 5), _components(F(" -3.2 ")))
187 self.assertEqual((-3, 1), _components(F(" -3. ")))
188 self.assertEqual((3, 5), _components(F(" .6 ")))
189 self.assertEqual((1, 3125), _components(F("32.e-5")))
190 self.assertEqual((1000000, 1), _components(F("1E+06")))
191 self.assertEqual((-12300, 1), _components(F("-1.23e4")))
192 self.assertEqual((0, 1), _components(F(" .0e+0\t")))
193 self.assertEqual((0, 1), _components(F("-0.000e0")))
Christian Heimes587c2bf2008-01-19 16:21:02 +0000194
195 self.assertRaisesMessage(
Christian Heimes3feef612008-02-11 06:19:17 +0000196 ZeroDivisionError, "Fraction(3, 0)",
Christian Heimes68f5fbe2008-02-14 08:27:37 +0000197 F, "3/0")
Christian Heimes587c2bf2008-01-19 16:21:02 +0000198 self.assertRaisesMessage(
Benjamin Petersondcf97b92008-07-02 17:30:14 +0000199 ValueError, "Invalid literal for Fraction: '3/'",
Christian Heimes68f5fbe2008-02-14 08:27:37 +0000200 F, "3/")
Christian Heimes587c2bf2008-01-19 16:21:02 +0000201 self.assertRaisesMessage(
Mark Dickinsoncf63f2f2009-04-22 17:50:21 +0000202 ValueError, "Invalid literal for Fraction: '/2'",
203 F, "/2")
204 self.assertRaisesMessage(
Benjamin Petersondcf97b92008-07-02 17:30:14 +0000205 ValueError, "Invalid literal for Fraction: '3 /2'",
Christian Heimes68f5fbe2008-02-14 08:27:37 +0000206 F, "3 /2")
Christian Heimes587c2bf2008-01-19 16:21:02 +0000207 self.assertRaisesMessage(
208 # Denominators don't need a sign.
Benjamin Petersondcf97b92008-07-02 17:30:14 +0000209 ValueError, "Invalid literal for Fraction: '3/+2'",
Christian Heimes68f5fbe2008-02-14 08:27:37 +0000210 F, "3/+2")
Christian Heimes587c2bf2008-01-19 16:21:02 +0000211 self.assertRaisesMessage(
212 # Imitate float's parsing.
Benjamin Petersondcf97b92008-07-02 17:30:14 +0000213 ValueError, "Invalid literal for Fraction: '+ 3/2'",
Christian Heimes68f5fbe2008-02-14 08:27:37 +0000214 F, "+ 3/2")
Christian Heimes587c2bf2008-01-19 16:21:02 +0000215 self.assertRaisesMessage(
Christian Heimesaf98da12008-01-27 15:18:18 +0000216 # Avoid treating '.' as a regex special character.
Benjamin Petersondcf97b92008-07-02 17:30:14 +0000217 ValueError, "Invalid literal for Fraction: '3a2'",
Christian Heimes68f5fbe2008-02-14 08:27:37 +0000218 F, "3a2")
Christian Heimesaf98da12008-01-27 15:18:18 +0000219 self.assertRaisesMessage(
Christian Heimesaf98da12008-01-27 15:18:18 +0000220 # Don't accept combinations of decimals and rationals.
Benjamin Petersondcf97b92008-07-02 17:30:14 +0000221 ValueError, "Invalid literal for Fraction: '3/7.2'",
Christian Heimes68f5fbe2008-02-14 08:27:37 +0000222 F, "3/7.2")
Christian Heimesaf98da12008-01-27 15:18:18 +0000223 self.assertRaisesMessage(
224 # Don't accept combinations of decimals and rationals.
Benjamin Petersondcf97b92008-07-02 17:30:14 +0000225 ValueError, "Invalid literal for Fraction: '3.2/7'",
Christian Heimes68f5fbe2008-02-14 08:27:37 +0000226 F, "3.2/7")
Christian Heimes292d3512008-02-03 16:51:08 +0000227 self.assertRaisesMessage(
228 # Allow 3. and .3, but not .
Benjamin Petersondcf97b92008-07-02 17:30:14 +0000229 ValueError, "Invalid literal for Fraction: '.'",
Christian Heimes68f5fbe2008-02-14 08:27:37 +0000230 F, ".")
Christian Heimes587c2bf2008-01-19 16:21:02 +0000231
232 def testImmutable(self):
Christian Heimes68f5fbe2008-02-14 08:27:37 +0000233 r = F(7, 3)
Christian Heimes587c2bf2008-01-19 16:21:02 +0000234 r.__init__(2, 15)
Ezio Melottib3aedd42010-11-20 19:04:17 +0000235 self.assertEqual((7, 3), _components(r))
Christian Heimes587c2bf2008-01-19 16:21:02 +0000236
Christian Heimes400adb02008-02-01 08:12:03 +0000237 self.assertRaises(AttributeError, setattr, r, 'numerator', 12)
238 self.assertRaises(AttributeError, setattr, r, 'denominator', 6)
Ezio Melottib3aedd42010-11-20 19:04:17 +0000239 self.assertEqual((7, 3), _components(r))
Christian Heimes400adb02008-02-01 08:12:03 +0000240
241 # But if you _really_ need to:
242 r._numerator = 4
243 r._denominator = 2
Ezio Melottib3aedd42010-11-20 19:04:17 +0000244 self.assertEqual((4, 2), _components(r))
Christian Heimes400adb02008-02-01 08:12:03 +0000245 # Which breaks some important operations:
Ezio Melottib3aedd42010-11-20 19:04:17 +0000246 self.assertNotEqual(F(4, 2), r)
Christian Heimes400adb02008-02-01 08:12:03 +0000247
Guido van Rossum7736b5b2008-01-15 21:44:53 +0000248 def testFromFloat(self):
Georg Brandl2ee470f2008-07-16 12:55:28 +0000249 self.assertRaises(TypeError, F.from_float, 3+4j)
Ezio Melottib3aedd42010-11-20 19:04:17 +0000250 self.assertEqual((10, 1), _components(F.from_float(10)))
Georg Brandl3a9b0622009-01-03 22:07:57 +0000251 bigint = 1234567890123456789
Ezio Melottib3aedd42010-11-20 19:04:17 +0000252 self.assertEqual((bigint, 1), _components(F.from_float(bigint)))
253 self.assertEqual((0, 1), _components(F.from_float(-0.0)))
254 self.assertEqual((10, 1), _components(F.from_float(10.0)))
255 self.assertEqual((-5, 2), _components(F.from_float(-2.5)))
256 self.assertEqual((99999999999999991611392, 1),
257 _components(F.from_float(1e23)))
258 self.assertEqual(float(10**23), float(F.from_float(1e23)))
259 self.assertEqual((3602879701896397, 1125899906842624),
260 _components(F.from_float(3.2)))
261 self.assertEqual(3.2, float(F.from_float(3.2)))
Guido van Rossum7736b5b2008-01-15 21:44:53 +0000262
263 inf = 1e1000
264 nan = inf - inf
Mark Dickinson73726aa2012-11-15 20:58:40 +0000265 # bug 16469: error types should be consistent with float -> int
Guido van Rossum7736b5b2008-01-15 21:44:53 +0000266 self.assertRaisesMessage(
Serhiy Storchaka0d250bc2015-12-29 22:34:23 +0200267 OverflowError, "cannot convert Infinity to integer ratio",
Christian Heimes68f5fbe2008-02-14 08:27:37 +0000268 F.from_float, inf)
Guido van Rossum7736b5b2008-01-15 21:44:53 +0000269 self.assertRaisesMessage(
Serhiy Storchaka0d250bc2015-12-29 22:34:23 +0200270 OverflowError, "cannot convert Infinity to integer ratio",
Christian Heimes68f5fbe2008-02-14 08:27:37 +0000271 F.from_float, -inf)
Guido van Rossum7736b5b2008-01-15 21:44:53 +0000272 self.assertRaisesMessage(
Serhiy Storchaka0d250bc2015-12-29 22:34:23 +0200273 ValueError, "cannot convert NaN to integer ratio",
Christian Heimes68f5fbe2008-02-14 08:27:37 +0000274 F.from_float, nan)
Guido van Rossum7736b5b2008-01-15 21:44:53 +0000275
Christian Heimes587c2bf2008-01-19 16:21:02 +0000276 def testFromDecimal(self):
Georg Brandl2ee470f2008-07-16 12:55:28 +0000277 self.assertRaises(TypeError, F.from_decimal, 3+4j)
Ezio Melottib3aedd42010-11-20 19:04:17 +0000278 self.assertEqual(F(10, 1), F.from_decimal(10))
279 self.assertEqual(F(0), F.from_decimal(Decimal("-0")))
280 self.assertEqual(F(5, 10), F.from_decimal(Decimal("0.5")))
281 self.assertEqual(F(5, 1000), F.from_decimal(Decimal("5e-3")))
282 self.assertEqual(F(5000), F.from_decimal(Decimal("5e3")))
283 self.assertEqual(1 - F(1, 10**30),
284 F.from_decimal(Decimal("0." + "9" * 30)))
Christian Heimes587c2bf2008-01-19 16:21:02 +0000285
Mark Dickinson73726aa2012-11-15 20:58:40 +0000286 # bug 16469: error types should be consistent with decimal -> int
Christian Heimes587c2bf2008-01-19 16:21:02 +0000287 self.assertRaisesMessage(
Serhiy Storchaka0d250bc2015-12-29 22:34:23 +0200288 OverflowError, "cannot convert Infinity to integer ratio",
Christian Heimes68f5fbe2008-02-14 08:27:37 +0000289 F.from_decimal, Decimal("inf"))
Christian Heimes587c2bf2008-01-19 16:21:02 +0000290 self.assertRaisesMessage(
Serhiy Storchaka0d250bc2015-12-29 22:34:23 +0200291 OverflowError, "cannot convert Infinity to integer ratio",
Christian Heimes68f5fbe2008-02-14 08:27:37 +0000292 F.from_decimal, Decimal("-inf"))
Christian Heimes587c2bf2008-01-19 16:21:02 +0000293 self.assertRaisesMessage(
Serhiy Storchaka0d250bc2015-12-29 22:34:23 +0200294 ValueError, "cannot convert NaN to integer ratio",
Christian Heimes68f5fbe2008-02-14 08:27:37 +0000295 F.from_decimal, Decimal("nan"))
Christian Heimes587c2bf2008-01-19 16:21:02 +0000296 self.assertRaisesMessage(
Serhiy Storchaka0d250bc2015-12-29 22:34:23 +0200297 ValueError, "cannot convert NaN to integer ratio",
Christian Heimes68f5fbe2008-02-14 08:27:37 +0000298 F.from_decimal, Decimal("snan"))
Christian Heimes587c2bf2008-01-19 16:21:02 +0000299
Christian Heimes68f5fbe2008-02-14 08:27:37 +0000300 def testLimitDenominator(self):
301 rpi = F('3.1415926535897932')
302 self.assertEqual(rpi.limit_denominator(10000), F(355, 113))
303 self.assertEqual(-rpi.limit_denominator(10000), F(-355, 113))
304 self.assertEqual(rpi.limit_denominator(113), F(355, 113))
305 self.assertEqual(rpi.limit_denominator(112), F(333, 106))
306 self.assertEqual(F(201, 200).limit_denominator(100), F(1))
307 self.assertEqual(F(201, 200).limit_denominator(101), F(102, 101))
308 self.assertEqual(F(0).limit_denominator(10000), F(0))
Ezio Melotti682d3742012-02-29 14:05:53 +0200309 for i in (0, -1):
310 self.assertRaisesMessage(
311 ValueError, "max_denominator should be at least 1",
312 F(1).limit_denominator, i)
Christian Heimesbbffeb62008-01-24 09:42:52 +0000313
Guido van Rossum7736b5b2008-01-15 21:44:53 +0000314 def testConversions(self):
Christian Heimes68f5fbe2008-02-14 08:27:37 +0000315 self.assertTypedEquals(-1, math.trunc(F(-11, 10)))
Ezio Melotti682d3742012-02-29 14:05:53 +0200316 self.assertTypedEquals(1, math.trunc(F(11, 10)))
Christian Heimes68f5fbe2008-02-14 08:27:37 +0000317 self.assertTypedEquals(-2, math.floor(F(-11, 10)))
318 self.assertTypedEquals(-1, math.ceil(F(-11, 10)))
319 self.assertTypedEquals(-1, math.ceil(F(-10, 10)))
320 self.assertTypedEquals(-1, int(F(-11, 10)))
321 self.assertTypedEquals(0, round(F(-1, 10)))
322 self.assertTypedEquals(0, round(F(-5, 10)))
323 self.assertTypedEquals(-2, round(F(-15, 10)))
324 self.assertTypedEquals(-1, round(F(-7, 10)))
Guido van Rossum7736b5b2008-01-15 21:44:53 +0000325
Ezio Melottib3aedd42010-11-20 19:04:17 +0000326 self.assertEqual(False, bool(F(0, 1)))
327 self.assertEqual(True, bool(F(3, 2)))
Christian Heimes68f5fbe2008-02-14 08:27:37 +0000328 self.assertTypedEquals(0.1, float(F(1, 10)))
Guido van Rossum7736b5b2008-01-15 21:44:53 +0000329
330 # Check that __float__ isn't implemented by converting the
331 # numerator and denominator to float before dividing.
332 self.assertRaises(OverflowError, float, int('2'*400+'7'))
Ezio Melottib3aedd42010-11-20 19:04:17 +0000333 self.assertAlmostEqual(2.0/3,
334 float(F(int('2'*400+'7'), int('3'*400+'1'))))
Guido van Rossum7736b5b2008-01-15 21:44:53 +0000335
Christian Heimes68f5fbe2008-02-14 08:27:37 +0000336 self.assertTypedEquals(0.1+0j, complex(F(1,10)))
Guido van Rossum7736b5b2008-01-15 21:44:53 +0000337
338 def testRound(self):
Christian Heimes68f5fbe2008-02-14 08:27:37 +0000339 self.assertTypedEquals(F(-200), round(F(-150), -2))
340 self.assertTypedEquals(F(-200), round(F(-250), -2))
341 self.assertTypedEquals(F(30), round(F(26), -1))
342 self.assertTypedEquals(F(-2, 10), round(F(-15, 100), 1))
343 self.assertTypedEquals(F(-2, 10), round(F(-25, 100), 1))
Guido van Rossum7736b5b2008-01-15 21:44:53 +0000344
Guido van Rossum7736b5b2008-01-15 21:44:53 +0000345 def testArithmetic(self):
Ezio Melottib3aedd42010-11-20 19:04:17 +0000346 self.assertEqual(F(1, 2), F(1, 10) + F(2, 5))
347 self.assertEqual(F(-3, 10), F(1, 10) - F(2, 5))
348 self.assertEqual(F(1, 25), F(1, 10) * F(2, 5))
349 self.assertEqual(F(1, 4), F(1, 10) / F(2, 5))
Christian Heimes68f5fbe2008-02-14 08:27:37 +0000350 self.assertTypedEquals(2, F(9, 10) // F(2, 5))
351 self.assertTypedEquals(10**23, F(10**23, 1) // F(1))
Ezio Melottib3aedd42010-11-20 19:04:17 +0000352 self.assertEqual(F(2, 3), F(-7, 3) % F(3, 2))
353 self.assertEqual(F(8, 27), F(2, 3) ** F(3))
354 self.assertEqual(F(27, 8), F(2, 3) ** F(-3))
Christian Heimes68f5fbe2008-02-14 08:27:37 +0000355 self.assertTypedEquals(2.0, F(4) ** F(1, 2))
Ezio Melotti682d3742012-02-29 14:05:53 +0200356 self.assertEqual(F(1, 1), +F(1, 1))
Christian Heimes68f5fbe2008-02-14 08:27:37 +0000357 z = pow(F(-1), F(1, 2))
Ezio Melottib3aedd42010-11-20 19:04:17 +0000358 self.assertAlmostEqual(z.real, 0)
359 self.assertEqual(z.imag, 1)
Mark Dickinson84479652016-08-22 10:50:53 +0100360 # Regression test for #27539.
361 p = F(-1, 2) ** 0
362 self.assertEqual(p, F(1, 1))
363 self.assertEqual(p.numerator, 1)
364 self.assertEqual(p.denominator, 1)
365 p = F(-1, 2) ** -1
366 self.assertEqual(p, F(-2, 1))
367 self.assertEqual(p.numerator, -2)
368 self.assertEqual(p.denominator, 1)
369 p = F(-1, 2) ** -2
370 self.assertEqual(p, F(4, 1))
371 self.assertEqual(p.numerator, 4)
372 self.assertEqual(p.denominator, 1)
Guido van Rossum7736b5b2008-01-15 21:44:53 +0000373
374 def testMixedArithmetic(self):
Christian Heimes68f5fbe2008-02-14 08:27:37 +0000375 self.assertTypedEquals(F(11, 10), F(1, 10) + 1)
376 self.assertTypedEquals(1.1, F(1, 10) + 1.0)
377 self.assertTypedEquals(1.1 + 0j, F(1, 10) + (1.0 + 0j))
378 self.assertTypedEquals(F(11, 10), 1 + F(1, 10))
379 self.assertTypedEquals(1.1, 1.0 + F(1, 10))
380 self.assertTypedEquals(1.1 + 0j, (1.0 + 0j) + F(1, 10))
Guido van Rossum7736b5b2008-01-15 21:44:53 +0000381
Christian Heimes68f5fbe2008-02-14 08:27:37 +0000382 self.assertTypedEquals(F(-9, 10), F(1, 10) - 1)
383 self.assertTypedEquals(-0.9, F(1, 10) - 1.0)
384 self.assertTypedEquals(-0.9 + 0j, F(1, 10) - (1.0 + 0j))
385 self.assertTypedEquals(F(9, 10), 1 - F(1, 10))
386 self.assertTypedEquals(0.9, 1.0 - F(1, 10))
387 self.assertTypedEquals(0.9 + 0j, (1.0 + 0j) - F(1, 10))
Guido van Rossum7736b5b2008-01-15 21:44:53 +0000388
Christian Heimes68f5fbe2008-02-14 08:27:37 +0000389 self.assertTypedEquals(F(1, 10), F(1, 10) * 1)
390 self.assertTypedEquals(0.1, F(1, 10) * 1.0)
391 self.assertTypedEquals(0.1 + 0j, F(1, 10) * (1.0 + 0j))
392 self.assertTypedEquals(F(1, 10), 1 * F(1, 10))
393 self.assertTypedEquals(0.1, 1.0 * F(1, 10))
394 self.assertTypedEquals(0.1 + 0j, (1.0 + 0j) * F(1, 10))
Guido van Rossum7736b5b2008-01-15 21:44:53 +0000395
Christian Heimes68f5fbe2008-02-14 08:27:37 +0000396 self.assertTypedEquals(F(1, 10), F(1, 10) / 1)
397 self.assertTypedEquals(0.1, F(1, 10) / 1.0)
398 self.assertTypedEquals(0.1 + 0j, F(1, 10) / (1.0 + 0j))
399 self.assertTypedEquals(F(10, 1), 1 / F(1, 10))
400 self.assertTypedEquals(10.0, 1.0 / F(1, 10))
401 self.assertTypedEquals(10.0 + 0j, (1.0 + 0j) / F(1, 10))
Guido van Rossum7736b5b2008-01-15 21:44:53 +0000402
Christian Heimes68f5fbe2008-02-14 08:27:37 +0000403 self.assertTypedEquals(0, F(1, 10) // 1)
Elias Zamaria393f1ff2018-08-26 23:59:28 -0700404 self.assertTypedEquals(0.0, F(1, 10) // 1.0)
Christian Heimes68f5fbe2008-02-14 08:27:37 +0000405 self.assertTypedEquals(10, 1 // F(1, 10))
406 self.assertTypedEquals(10**23, 10**22 // F(1, 10))
Elias Zamaria393f1ff2018-08-26 23:59:28 -0700407 self.assertTypedEquals(1.0 // 0.1, 1.0 // F(1, 10))
Guido van Rossum7736b5b2008-01-15 21:44:53 +0000408
Christian Heimes68f5fbe2008-02-14 08:27:37 +0000409 self.assertTypedEquals(F(1, 10), F(1, 10) % 1)
410 self.assertTypedEquals(0.1, F(1, 10) % 1.0)
411 self.assertTypedEquals(F(0, 1), 1 % F(1, 10))
Elias Zamaria393f1ff2018-08-26 23:59:28 -0700412 self.assertTypedEquals(1.0 % 0.1, 1.0 % F(1, 10))
413 self.assertTypedEquals(0.1, F(1, 10) % float('inf'))
414 self.assertTypedEquals(float('-inf'), F(1, 10) % float('-inf'))
415 self.assertTypedEquals(float('inf'), F(-1, 10) % float('inf'))
416 self.assertTypedEquals(-0.1, F(-1, 10) % float('-inf'))
Guido van Rossum7736b5b2008-01-15 21:44:53 +0000417
418 # No need for divmod since we don't override it.
419
420 # ** has more interesting conversion rules.
Christian Heimes68f5fbe2008-02-14 08:27:37 +0000421 self.assertTypedEquals(F(100, 1), F(1, 10) ** -2)
422 self.assertTypedEquals(F(100, 1), F(10, 1) ** 2)
423 self.assertTypedEquals(0.1, F(1, 10) ** 1.0)
424 self.assertTypedEquals(0.1 + 0j, F(1, 10) ** (1.0 + 0j))
425 self.assertTypedEquals(4 , 2 ** F(2, 1))
426 z = pow(-1, F(1, 2))
Ezio Melottib3aedd42010-11-20 19:04:17 +0000427 self.assertAlmostEqual(0, z.real)
428 self.assertEqual(1, z.imag)
Christian Heimes68f5fbe2008-02-14 08:27:37 +0000429 self.assertTypedEquals(F(1, 4) , 2 ** F(-2, 1))
430 self.assertTypedEquals(2.0 , 4 ** F(1, 2))
431 self.assertTypedEquals(0.25, 2.0 ** F(-2, 1))
432 self.assertTypedEquals(1.0 + 0j, (1.0 + 0j) ** F(1, 10))
Mark Dickinson3c286e22014-04-05 09:29:00 +0100433 self.assertRaises(ZeroDivisionError, operator.pow,
434 F(0, 1), -2)
Guido van Rossum7736b5b2008-01-15 21:44:53 +0000435
436 def testMixingWithDecimal(self):
Mark Dickinson08ade6f2010-06-11 10:44:52 +0000437 # Decimal refuses mixed arithmetic (but not mixed comparisons)
Stefan Krah1919b7e2012-03-21 18:25:23 +0100438 self.assertRaises(TypeError, operator.add,
439 F(3,11), Decimal('3.1415926'))
440 self.assertRaises(TypeError, operator.add,
441 Decimal('3.1415926'), F(3,11))
Guido van Rossum7736b5b2008-01-15 21:44:53 +0000442
443 def testComparisons(self):
Christian Heimes68f5fbe2008-02-14 08:27:37 +0000444 self.assertTrue(F(1, 2) < F(2, 3))
445 self.assertFalse(F(1, 2) < F(1, 2))
446 self.assertTrue(F(1, 2) <= F(2, 3))
447 self.assertTrue(F(1, 2) <= F(1, 2))
448 self.assertFalse(F(2, 3) <= F(1, 2))
449 self.assertTrue(F(1, 2) == F(1, 2))
450 self.assertFalse(F(1, 2) == F(1, 3))
451 self.assertFalse(F(1, 2) != F(1, 2))
452 self.assertTrue(F(1, 2) != F(1, 3))
Guido van Rossum7736b5b2008-01-15 21:44:53 +0000453
Mark Dickinson85424c92009-07-18 14:41:42 +0000454 def testComparisonsDummyRational(self):
455 self.assertTrue(F(1, 2) == DummyRational(1, 2))
456 self.assertTrue(DummyRational(1, 2) == F(1, 2))
457 self.assertFalse(F(1, 2) == DummyRational(3, 4))
458 self.assertFalse(DummyRational(3, 4) == F(1, 2))
459
460 self.assertTrue(F(1, 2) < DummyRational(3, 4))
461 self.assertFalse(F(1, 2) < DummyRational(1, 2))
462 self.assertFalse(F(1, 2) < DummyRational(1, 7))
463 self.assertFalse(F(1, 2) > DummyRational(3, 4))
464 self.assertFalse(F(1, 2) > DummyRational(1, 2))
465 self.assertTrue(F(1, 2) > DummyRational(1, 7))
466 self.assertTrue(F(1, 2) <= DummyRational(3, 4))
467 self.assertTrue(F(1, 2) <= DummyRational(1, 2))
468 self.assertFalse(F(1, 2) <= DummyRational(1, 7))
469 self.assertFalse(F(1, 2) >= DummyRational(3, 4))
470 self.assertTrue(F(1, 2) >= DummyRational(1, 2))
471 self.assertTrue(F(1, 2) >= DummyRational(1, 7))
472
473 self.assertTrue(DummyRational(1, 2) < F(3, 4))
474 self.assertFalse(DummyRational(1, 2) < F(1, 2))
475 self.assertFalse(DummyRational(1, 2) < F(1, 7))
476 self.assertFalse(DummyRational(1, 2) > F(3, 4))
477 self.assertFalse(DummyRational(1, 2) > F(1, 2))
478 self.assertTrue(DummyRational(1, 2) > F(1, 7))
479 self.assertTrue(DummyRational(1, 2) <= F(3, 4))
480 self.assertTrue(DummyRational(1, 2) <= F(1, 2))
481 self.assertFalse(DummyRational(1, 2) <= F(1, 7))
482 self.assertFalse(DummyRational(1, 2) >= F(3, 4))
483 self.assertTrue(DummyRational(1, 2) >= F(1, 2))
484 self.assertTrue(DummyRational(1, 2) >= F(1, 7))
485
486 def testComparisonsDummyFloat(self):
487 x = DummyFloat(1./3.)
488 y = F(1, 3)
489 self.assertTrue(x != y)
490 self.assertTrue(x < y or x > y)
491 self.assertFalse(x == y)
492 self.assertFalse(x <= y and x >= y)
493 self.assertTrue(y != x)
494 self.assertTrue(y < x or y > x)
495 self.assertFalse(y == x)
496 self.assertFalse(y <= x and y >= x)
497
Guido van Rossum7736b5b2008-01-15 21:44:53 +0000498 def testMixedLess(self):
Christian Heimes68f5fbe2008-02-14 08:27:37 +0000499 self.assertTrue(2 < F(5, 2))
500 self.assertFalse(2 < F(4, 2))
501 self.assertTrue(F(5, 2) < 3)
502 self.assertFalse(F(4, 2) < 2)
Guido van Rossum7736b5b2008-01-15 21:44:53 +0000503
Christian Heimes68f5fbe2008-02-14 08:27:37 +0000504 self.assertTrue(F(1, 2) < 0.6)
505 self.assertFalse(F(1, 2) < 0.4)
506 self.assertTrue(0.4 < F(1, 2))
507 self.assertFalse(0.5 < F(1, 2))
Guido van Rossum7736b5b2008-01-15 21:44:53 +0000508
Mark Dickinson85424c92009-07-18 14:41:42 +0000509 self.assertFalse(float('inf') < F(1, 2))
510 self.assertTrue(float('-inf') < F(0, 10))
511 self.assertFalse(float('nan') < F(-3, 7))
512 self.assertTrue(F(1, 2) < float('inf'))
513 self.assertFalse(F(17, 12) < float('-inf'))
514 self.assertFalse(F(144, -89) < float('nan'))
515
Guido van Rossum7736b5b2008-01-15 21:44:53 +0000516 def testMixedLessEqual(self):
Christian Heimes68f5fbe2008-02-14 08:27:37 +0000517 self.assertTrue(0.5 <= F(1, 2))
518 self.assertFalse(0.6 <= F(1, 2))
519 self.assertTrue(F(1, 2) <= 0.5)
520 self.assertFalse(F(1, 2) <= 0.4)
521 self.assertTrue(2 <= F(4, 2))
522 self.assertFalse(2 <= F(3, 2))
523 self.assertTrue(F(4, 2) <= 2)
524 self.assertFalse(F(5, 2) <= 2)
Guido van Rossum7736b5b2008-01-15 21:44:53 +0000525
Mark Dickinson85424c92009-07-18 14:41:42 +0000526 self.assertFalse(float('inf') <= F(1, 2))
527 self.assertTrue(float('-inf') <= F(0, 10))
528 self.assertFalse(float('nan') <= F(-3, 7))
529 self.assertTrue(F(1, 2) <= float('inf'))
530 self.assertFalse(F(17, 12) <= float('-inf'))
531 self.assertFalse(F(144, -89) <= float('nan'))
532
Guido van Rossum7736b5b2008-01-15 21:44:53 +0000533 def testBigFloatComparisons(self):
534 # Because 10**23 can't be represented exactly as a float:
Christian Heimes68f5fbe2008-02-14 08:27:37 +0000535 self.assertFalse(F(10**23) == float(10**23))
Guido van Rossum7736b5b2008-01-15 21:44:53 +0000536 # The first test demonstrates why these are important.
Christian Heimes68f5fbe2008-02-14 08:27:37 +0000537 self.assertFalse(1e23 < float(F(math.trunc(1e23) + 1)))
538 self.assertTrue(1e23 < F(math.trunc(1e23) + 1))
539 self.assertFalse(1e23 <= F(math.trunc(1e23) - 1))
540 self.assertTrue(1e23 > F(math.trunc(1e23) - 1))
541 self.assertFalse(1e23 >= F(math.trunc(1e23) + 1))
Guido van Rossum7736b5b2008-01-15 21:44:53 +0000542
543 def testBigComplexComparisons(self):
Christian Heimes68f5fbe2008-02-14 08:27:37 +0000544 self.assertFalse(F(10**23) == complex(10**23))
Mark Dickinson327f02c2010-03-27 11:11:13 +0000545 self.assertRaises(TypeError, operator.gt, F(10**23), complex(10**23))
546 self.assertRaises(TypeError, operator.le, F(10**23), complex(10**23))
547
548 x = F(3, 8)
549 z = complex(0.375, 0.0)
550 w = complex(0.375, 0.2)
551 self.assertTrue(x == z)
552 self.assertFalse(x != z)
553 self.assertFalse(x == w)
554 self.assertTrue(x != w)
555 for op in operator.lt, operator.le, operator.gt, operator.ge:
556 self.assertRaises(TypeError, op, x, z)
557 self.assertRaises(TypeError, op, z, x)
558 self.assertRaises(TypeError, op, x, w)
559 self.assertRaises(TypeError, op, w, x)
Guido van Rossum7736b5b2008-01-15 21:44:53 +0000560
561 def testMixedEqual(self):
Christian Heimes68f5fbe2008-02-14 08:27:37 +0000562 self.assertTrue(0.5 == F(1, 2))
563 self.assertFalse(0.6 == F(1, 2))
564 self.assertTrue(F(1, 2) == 0.5)
565 self.assertFalse(F(1, 2) == 0.4)
566 self.assertTrue(2 == F(4, 2))
567 self.assertFalse(2 == F(3, 2))
568 self.assertTrue(F(4, 2) == 2)
569 self.assertFalse(F(5, 2) == 2)
Mark Dickinson85424c92009-07-18 14:41:42 +0000570 self.assertFalse(F(5, 2) == float('nan'))
571 self.assertFalse(float('nan') == F(3, 7))
572 self.assertFalse(F(5, 2) == float('inf'))
573 self.assertFalse(float('-inf') == F(2, 5))
Guido van Rossum7736b5b2008-01-15 21:44:53 +0000574
575 def testStringification(self):
Ezio Melottib3aedd42010-11-20 19:04:17 +0000576 self.assertEqual("Fraction(7, 3)", repr(F(7, 3)))
577 self.assertEqual("Fraction(6283185307, 2000000000)",
578 repr(F('3.1415926535')))
579 self.assertEqual("Fraction(-1, 100000000000000000000)",
580 repr(F(1, -10**20)))
581 self.assertEqual("7/3", str(F(7, 3)))
582 self.assertEqual("7", str(F(7, 1)))
Guido van Rossum7736b5b2008-01-15 21:44:53 +0000583
584 def testHash(self):
Ezio Melotti682d3742012-02-29 14:05:53 +0200585 hmod = sys.hash_info.modulus
586 hinf = sys.hash_info.inf
Ezio Melottib3aedd42010-11-20 19:04:17 +0000587 self.assertEqual(hash(2.5), hash(F(5, 2)))
588 self.assertEqual(hash(10**50), hash(F(10**50)))
589 self.assertNotEqual(hash(float(10**23)), hash(F(10**23)))
Ezio Melotti682d3742012-02-29 14:05:53 +0200590 self.assertEqual(hinf, hash(F(1, hmod)))
Mark Dickinsonfec66202010-11-13 10:27:38 +0000591 # Check that __hash__ produces the same value as hash(), for
592 # consistency with int and Decimal. (See issue #10356.)
Ezio Melottib3aedd42010-11-20 19:04:17 +0000593 self.assertEqual(hash(F(-1)), F(-1).__hash__())
Guido van Rossum7736b5b2008-01-15 21:44:53 +0000594
595 def testApproximatePi(self):
596 # Algorithm borrowed from
597 # http://docs.python.org/lib/decimal-recipes.html
Christian Heimes68f5fbe2008-02-14 08:27:37 +0000598 three = F(3)
Guido van Rossum7736b5b2008-01-15 21:44:53 +0000599 lasts, t, s, n, na, d, da = 0, three, 3, 1, 0, 0, 24
Christian Heimes68f5fbe2008-02-14 08:27:37 +0000600 while abs(s - lasts) > F(1, 10**9):
Guido van Rossum7736b5b2008-01-15 21:44:53 +0000601 lasts = s
602 n, na = n+na, na+8
603 d, da = d+da, da+32
604 t = (t * n) / d
605 s += t
Ezio Melottib3aedd42010-11-20 19:04:17 +0000606 self.assertAlmostEqual(math.pi, s)
Guido van Rossum7736b5b2008-01-15 21:44:53 +0000607
608 def testApproximateCos1(self):
609 # Algorithm borrowed from
610 # http://docs.python.org/lib/decimal-recipes.html
Christian Heimes68f5fbe2008-02-14 08:27:37 +0000611 x = F(1)
612 i, lasts, s, fact, num, sign = 0, 0, F(1), 1, 1, 1
613 while abs(s - lasts) > F(1, 10**9):
Guido van Rossum7736b5b2008-01-15 21:44:53 +0000614 lasts = s
615 i += 2
616 fact *= i * (i-1)
617 num *= x * x
618 sign *= -1
619 s += num / fact * sign
Ezio Melottib3aedd42010-11-20 19:04:17 +0000620 self.assertAlmostEqual(math.cos(1), s)
Guido van Rossum7736b5b2008-01-15 21:44:53 +0000621
Christian Heimes969fe572008-01-25 11:23:10 +0000622 def test_copy_deepcopy_pickle(self):
Christian Heimes68f5fbe2008-02-14 08:27:37 +0000623 r = F(13, 7)
Ezio Melotti682d3742012-02-29 14:05:53 +0200624 dr = DummyFraction(13, 7)
Christian Heimes969fe572008-01-25 11:23:10 +0000625 self.assertEqual(r, loads(dumps(r)))
626 self.assertEqual(id(r), id(copy(r)))
627 self.assertEqual(id(r), id(deepcopy(r)))
Ezio Melotti682d3742012-02-29 14:05:53 +0200628 self.assertNotEqual(id(dr), id(copy(dr)))
629 self.assertNotEqual(id(dr), id(deepcopy(dr)))
630 self.assertTypedEquals(dr, copy(dr))
631 self.assertTypedEquals(dr, deepcopy(dr))
Christian Heimes969fe572008-01-25 11:23:10 +0000632
Mark Dickinsonc28ad272009-02-12 17:58:36 +0000633 def test_slots(self):
634 # Issue 4998
635 r = F(13, 7)
636 self.assertRaises(AttributeError, setattr, r, 'a', 10)
637
Guido van Rossum7736b5b2008-01-15 21:44:53 +0000638if __name__ == '__main__':
Zachary Ware38c707e2015-04-13 15:00:43 -0500639 unittest.main()