mirror of https://github.com/python/cpython
121 lines
3.9 KiB
Python
121 lines
3.9 KiB
Python
from enum import Enum, IntEnum
|
|
from math import isnan
|
|
from test.test_json import PyTest, CTest
|
|
|
|
SMALL = 1
|
|
BIG = 1<<32
|
|
HUGE = 1<<64
|
|
REALLY_HUGE = 1<<96
|
|
|
|
class BigNum(IntEnum):
|
|
small = SMALL
|
|
big = BIG
|
|
huge = HUGE
|
|
really_huge = REALLY_HUGE
|
|
|
|
E = 2.718281
|
|
PI = 3.141593
|
|
TAU = 2 * PI
|
|
|
|
class FloatNum(float, Enum):
|
|
e = E
|
|
pi = PI
|
|
tau = TAU
|
|
|
|
INF = float('inf')
|
|
NEG_INF = float('-inf')
|
|
NAN = float('nan')
|
|
|
|
class WierdNum(float, Enum):
|
|
inf = INF
|
|
neg_inf = NEG_INF
|
|
nan = NAN
|
|
|
|
class TestEnum:
|
|
|
|
def test_floats(self):
|
|
for enum in FloatNum:
|
|
self.assertEqual(self.dumps(enum), repr(enum.value))
|
|
self.assertEqual(float(self.dumps(enum)), enum)
|
|
self.assertEqual(self.loads(self.dumps(enum)), enum)
|
|
|
|
def test_weird_floats(self):
|
|
for enum, expected in zip(WierdNum, ('Infinity', '-Infinity', 'NaN')):
|
|
self.assertEqual(self.dumps(enum), expected)
|
|
if not isnan(enum):
|
|
self.assertEqual(float(self.dumps(enum)), enum)
|
|
self.assertEqual(self.loads(self.dumps(enum)), enum)
|
|
else:
|
|
self.assertTrue(isnan(float(self.dumps(enum))))
|
|
self.assertTrue(isnan(self.loads(self.dumps(enum))))
|
|
|
|
def test_ints(self):
|
|
for enum in BigNum:
|
|
self.assertEqual(self.dumps(enum), str(enum.value))
|
|
self.assertEqual(int(self.dumps(enum)), enum)
|
|
self.assertEqual(self.loads(self.dumps(enum)), enum)
|
|
|
|
def test_list(self):
|
|
self.assertEqual(self.dumps(list(BigNum)),
|
|
str([SMALL, BIG, HUGE, REALLY_HUGE]))
|
|
self.assertEqual(self.loads(self.dumps(list(BigNum))),
|
|
list(BigNum))
|
|
self.assertEqual(self.dumps(list(FloatNum)),
|
|
str([E, PI, TAU]))
|
|
self.assertEqual(self.loads(self.dumps(list(FloatNum))),
|
|
list(FloatNum))
|
|
self.assertEqual(self.dumps(list(WierdNum)),
|
|
'[Infinity, -Infinity, NaN]')
|
|
self.assertEqual(self.loads(self.dumps(list(WierdNum)))[:2],
|
|
list(WierdNum)[:2])
|
|
self.assertTrue(isnan(self.loads(self.dumps(list(WierdNum)))[2]))
|
|
|
|
def test_dict_keys(self):
|
|
s, b, h, r = BigNum
|
|
e, p, t = FloatNum
|
|
i, j, n = WierdNum
|
|
d = {
|
|
s:'tiny', b:'large', h:'larger', r:'largest',
|
|
e:"Euler's number", p:'pi', t:'tau',
|
|
i:'Infinity', j:'-Infinity', n:'NaN',
|
|
}
|
|
nd = self.loads(self.dumps(d))
|
|
self.assertEqual(nd[str(SMALL)], 'tiny')
|
|
self.assertEqual(nd[str(BIG)], 'large')
|
|
self.assertEqual(nd[str(HUGE)], 'larger')
|
|
self.assertEqual(nd[str(REALLY_HUGE)], 'largest')
|
|
self.assertEqual(nd[repr(E)], "Euler's number")
|
|
self.assertEqual(nd[repr(PI)], 'pi')
|
|
self.assertEqual(nd[repr(TAU)], 'tau')
|
|
self.assertEqual(nd['Infinity'], 'Infinity')
|
|
self.assertEqual(nd['-Infinity'], '-Infinity')
|
|
self.assertEqual(nd['NaN'], 'NaN')
|
|
|
|
def test_dict_values(self):
|
|
d = dict(
|
|
tiny=BigNum.small,
|
|
large=BigNum.big,
|
|
larger=BigNum.huge,
|
|
largest=BigNum.really_huge,
|
|
e=FloatNum.e,
|
|
pi=FloatNum.pi,
|
|
tau=FloatNum.tau,
|
|
i=WierdNum.inf,
|
|
j=WierdNum.neg_inf,
|
|
n=WierdNum.nan,
|
|
)
|
|
nd = self.loads(self.dumps(d))
|
|
self.assertEqual(nd['tiny'], SMALL)
|
|
self.assertEqual(nd['large'], BIG)
|
|
self.assertEqual(nd['larger'], HUGE)
|
|
self.assertEqual(nd['largest'], REALLY_HUGE)
|
|
self.assertEqual(nd['e'], E)
|
|
self.assertEqual(nd['pi'], PI)
|
|
self.assertEqual(nd['tau'], TAU)
|
|
self.assertEqual(nd['i'], INF)
|
|
self.assertEqual(nd['j'], NEG_INF)
|
|
self.assertTrue(isnan(nd['n']))
|
|
|
|
class TestPyEnum(TestEnum, PyTest): pass
|
|
class TestCEnum(TestEnum, CTest): pass
|