| import math |
| import sys |
| import unittest |
| import warnings |
| |
| from test.test_capi.test_getargs import (Float, FloatSubclass, FloatSubclass2, |
| BadIndex2, BadFloat2, Index, BadIndex, |
| BadFloat) |
| from test.support import import_helper |
| |
| _testcapi = import_helper.import_module('_testcapi') |
| _testlimitedcapi = import_helper.import_module('_testlimitedcapi') |
| |
| NULL = None |
| |
| # For PyFloat_Pack/Unpack* |
| BIG_ENDIAN = 0 |
| LITTLE_ENDIAN = 1 |
| EPSILON = { |
| 2: 2.0 ** -11, # binary16 |
| 4: 2.0 ** -24, # binary32 |
| 8: 2.0 ** -53, # binary64 |
| } |
| |
| HAVE_IEEE_754 = float.__getformat__("double").startswith("IEEE") |
| INF = float("inf") |
| NAN = float("nan") |
| |
| |
| class CAPIFloatTest(unittest.TestCase): |
| def test_check(self): |
| # Test PyFloat_Check() |
| check = _testlimitedcapi.float_check |
| |
| self.assertTrue(check(4.25)) |
| self.assertTrue(check(FloatSubclass(4.25))) |
| self.assertFalse(check(Float())) |
| self.assertFalse(check(3)) |
| self.assertFalse(check(object())) |
| |
| # CRASHES check(NULL) |
| |
| def test_checkexact(self): |
| # Test PyFloat_CheckExact() |
| checkexact = _testlimitedcapi.float_checkexact |
| |
| self.assertTrue(checkexact(4.25)) |
| self.assertFalse(checkexact(FloatSubclass(4.25))) |
| self.assertFalse(checkexact(Float())) |
| self.assertFalse(checkexact(3)) |
| self.assertFalse(checkexact(object())) |
| |
| # CRASHES checkexact(NULL) |
| |
| def test_fromstring(self): |
| # Test PyFloat_FromString() |
| fromstring = _testlimitedcapi.float_fromstring |
| |
| self.assertEqual(fromstring("4.25"), 4.25) |
| self.assertEqual(fromstring(b"4.25"), 4.25) |
| self.assertRaises(ValueError, fromstring, "4.25\0") |
| self.assertRaises(ValueError, fromstring, b"4.25\0") |
| |
| self.assertEqual(fromstring(bytearray(b"4.25")), 4.25) |
| |
| self.assertEqual(fromstring(memoryview(b"4.25")), 4.25) |
| self.assertEqual(fromstring(memoryview(b"4.255")[:-1]), 4.25) |
| self.assertRaises(TypeError, fromstring, memoryview(b"4.25")[::2]) |
| |
| self.assertRaises(TypeError, fromstring, 4.25) |
| |
| # CRASHES fromstring(NULL) |
| |
| def test_fromdouble(self): |
| # Test PyFloat_FromDouble() |
| fromdouble = _testlimitedcapi.float_fromdouble |
| |
| self.assertEqual(fromdouble(4.25), 4.25) |
| |
| def test_asdouble(self): |
| # Test PyFloat_AsDouble() |
| asdouble = _testlimitedcapi.float_asdouble |
| |
| class BadFloat3: |
| def __float__(self): |
| raise RuntimeError |
| |
| self.assertEqual(asdouble(4.25), 4.25) |
| self.assertEqual(asdouble(-1.0), -1.0) |
| self.assertEqual(asdouble(42), 42.0) |
| self.assertEqual(asdouble(-1), -1.0) |
| self.assertEqual(asdouble(2**1000), float(2**1000)) |
| |
| self.assertEqual(asdouble(FloatSubclass(4.25)), 4.25) |
| self.assertEqual(asdouble(FloatSubclass2(4.25)), 4.25) |
| self.assertEqual(asdouble(Index()), 99.) |
| |
| self.assertRaises(TypeError, asdouble, BadIndex()) |
| self.assertRaises(TypeError, asdouble, BadFloat()) |
| self.assertRaises(RuntimeError, asdouble, BadFloat3()) |
| with self.assertWarns(DeprecationWarning): |
| self.assertEqual(asdouble(BadIndex2()), 1.) |
| with self.assertWarns(DeprecationWarning): |
| self.assertEqual(asdouble(BadFloat2()), 4.25) |
| with warnings.catch_warnings(): |
| warnings.simplefilter("error", DeprecationWarning) |
| self.assertRaises(DeprecationWarning, asdouble, BadFloat2()) |
| self.assertRaises(TypeError, asdouble, object()) |
| self.assertRaises(TypeError, asdouble, NULL) |
| |
| def test_getinfo(self): |
| # Test PyFloat_GetInfo() |
| getinfo = _testlimitedcapi.float_getinfo |
| |
| self.assertEqual(getinfo(), sys.float_info) |
| |
| def test_getmax(self): |
| # Test PyFloat_GetMax() |
| getmax = _testlimitedcapi.float_getmax |
| |
| self.assertEqual(getmax(), sys.float_info.max) |
| |
| def test_getmin(self): |
| # Test PyFloat_GetMax() |
| getmin = _testlimitedcapi.float_getmin |
| |
| self.assertEqual(getmin(), sys.float_info.min) |
| |
| def test_pack(self): |
| # Test PyFloat_Pack2(), PyFloat_Pack4() and PyFloat_Pack8() |
| pack = _testcapi.float_pack |
| |
| self.assertEqual(pack(2, 1.5, BIG_ENDIAN), b'>\x00') |
| self.assertEqual(pack(4, 1.5, BIG_ENDIAN), b'?\xc0\x00\x00') |
| self.assertEqual(pack(8, 1.5, BIG_ENDIAN), |
| b'?\xf8\x00\x00\x00\x00\x00\x00') |
| self.assertEqual(pack(2, 1.5, LITTLE_ENDIAN), b'\x00>') |
| self.assertEqual(pack(4, 1.5, LITTLE_ENDIAN), b'\x00\x00\xc0?') |
| self.assertEqual(pack(8, 1.5, LITTLE_ENDIAN), |
| b'\x00\x00\x00\x00\x00\x00\xf8?') |
| |
| def test_unpack(self): |
| # Test PyFloat_Unpack2(), PyFloat_Unpack4() and PyFloat_Unpack8() |
| unpack = _testcapi.float_unpack |
| |
| self.assertEqual(unpack(b'>\x00', BIG_ENDIAN), 1.5) |
| self.assertEqual(unpack(b'?\xc0\x00\x00', BIG_ENDIAN), 1.5) |
| self.assertEqual(unpack(b'?\xf8\x00\x00\x00\x00\x00\x00', BIG_ENDIAN), |
| 1.5) |
| self.assertEqual(unpack(b'\x00>', LITTLE_ENDIAN), 1.5) |
| self.assertEqual(unpack(b'\x00\x00\xc0?', LITTLE_ENDIAN), 1.5) |
| self.assertEqual(unpack(b'\x00\x00\x00\x00\x00\x00\xf8?', LITTLE_ENDIAN), |
| 1.5) |
| |
| def test_pack_unpack_roundtrip(self): |
| pack = _testcapi.float_pack |
| unpack = _testcapi.float_unpack |
| |
| large = 2.0 ** 100 |
| values = [1.0, 1.5, large, 1.0/7, math.pi] |
| if HAVE_IEEE_754: |
| values.extend((INF, NAN)) |
| for value in values: |
| for size in (2, 4, 8,): |
| if size == 2 and value == large: |
| # too large for 16-bit float |
| continue |
| rel_tol = EPSILON[size] |
| for endian in (BIG_ENDIAN, LITTLE_ENDIAN): |
| with self.subTest(value=value, size=size, endian=endian): |
| data = pack(size, value, endian) |
| value2 = unpack(data, endian) |
| if math.isnan(value): |
| self.assertTrue(math.isnan(value2), (value, value2)) |
| elif size < 8: |
| self.assertTrue(math.isclose(value2, value, rel_tol=rel_tol), |
| (value, value2)) |
| else: |
| self.assertEqual(value2, value) |
| |
| |
| if __name__ == "__main__": |
| unittest.main() |
