1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
|
#!/usr/bin/env python
# Copyright (C) 2023 The Qt Company Ltd.
# SPDX-License-Identifier: LicenseRef-Qt-Commercial OR GPL-3.0-only WITH Qt-GPL-exception-1.0
'''Test cases for StdComplex class'''
import os
import math
import sys
import unittest
from pathlib import Path
sys.path.append(os.fspath(Path(__file__).resolve().parents[1]))
from shiboken_paths import init_paths
init_paths()
from sample import StdComplex
REAL = 5.0
IMAG = 2.3
class StdComplexTest(unittest.TestCase):
'''Test case for StdComplex class, exercising esoteric number
protocols (Py_nb_). For standard number protocols, see Point.'''
def testConversion(self):
pt = StdComplex(REAL, IMAG)
self.assertEqual(int(pt), int(round(pt.abs_value())))
self.assertEqual(float(pt), pt.abs_value())
def testAbs(self):
pt = StdComplex(REAL, IMAG)
self.assertEqual(abs(pt), pt.abs_value())
def testPow(self):
'''Compare pow() function to builtin Python type.'''
pt = StdComplex(REAL, IMAG)
result = pow(pt, StdComplex(2.0, 0))
py_pt = complex(REAL, IMAG)
py_result = pow(py_pt, complex(2.0, 0))
self.assertAlmostEqual(result.real(), py_result.real)
self.assertAlmostEqual(result.imag(), py_result.imag)
def testFloor(self):
pt = StdComplex(REAL, IMAG)
self.assertEqual(math.floor(pt), math.floor(pt.abs_value()))
def testCeil(self):
pt = StdComplex(REAL, IMAG)
self.assertEqual(math.ceil(pt), math.ceil(pt.abs_value()))
def testPlusOperator(self):
'''Test StdComplex class + operator.'''
pt1 = StdComplex(REAL, IMAG)
pt2 = StdComplex(0.5, 3.2)
self.assertEqual(pt1 + pt2, StdComplex(REAL + 0.5, IMAG + 3.2))
def testEqualOperator(self):
'''Test StdComplex class == operator.'''
pt1 = StdComplex(REAL, IMAG)
pt2 = StdComplex(REAL, IMAG)
pt3 = StdComplex(0.5, 3.2)
self.assertTrue(pt1 == pt1)
self.assertTrue(pt1 == pt2)
self.assertFalse(pt1 == pt3)
if __name__ == '__main__':
unittest.main()
|
