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
|
#!/usr/bin/env python
# Copyright (C) 2022 The Qt Company Ltd.
# SPDX-License-Identifier: LicenseRef-Qt-Commercial OR GPL-3.0-only WITH Qt-GPL-exception-1.0
'''Test cases for Collector class' shift operators.'''
import os
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 Collector, IntWrapper, ObjectType
class CollectorTest(unittest.TestCase):
'''Test cases for Collector class' shift operators.'''
def testLShiftOperatorSingleUse(self):
'''Test case for using the Collector.__lshift__ operator just one time.'''
collector = Collector()
collector << 13
self.assertEqual(collector.size(), 1)
self.assertEqual(collector.items(), [13])
def testLShiftOperatorMultipleUses(self):
'''Test case for using the Collector.__lshift__ operator many times in the same line.'''
collector = Collector()
collector << 2 << 3 << 5 << 7 << 11
self.assertEqual(collector.size(), 5)
self.assertEqual(collector.items(), [2, 3, 5, 7, 11])
class CollectorExternalOperator(unittest.TestCase):
'''Test cases for external operators of Collector'''
def testLShiftExternal(self):
'''Collector external operator'''
collector = Collector()
collector << IntWrapper(5)
self.assertEqual(collector.size(), 1)
self.assertEqual(collector.items(), [5])
class CollectorObjectType(unittest.TestCase):
'''Test cases for Collector ObjectType'''
def testBasic(self):
'''Collector << ObjectType # greedy collector'''
collector = Collector()
obj = ObjectType()
collector << obj
self.assertEqual(collector.items()[0], obj.identifier())
if __name__ == '__main__':
unittest.main()
|