summaryrefslogtreecommitdiffstats
path: root/Lib/test/test_capi.py
blob: 7196b7be0cfe0a42668284078efa5e88c2edfa7d (plain)
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
# Run the _testcapi module tests (tests for the Python/C API):  by defn,
# these are all functions _testcapi exports whose name begins with 'test_'.

import sys, unittest
from test import test_support
import _testcapi

for name in dir(_testcapi):
    if name.startswith('test_'):
        test = getattr(_testcapi, name)
        if test_support.verbose:
            print "internal", name
        try:
            test()
        except _testcapi.error:
            raise test_support.TestFailed, sys.exc_info()[1]

# some extra thread-state tests driven via _testcapi
def TestThreadState():
    import thread
    import time

    if test_support.verbose:
        print "auto-thread-state"

    idents = []

    def callback():
        idents.append(thread.get_ident())

    _testcapi._test_thread_state(callback)
    time.sleep(1)
    # Check our main thread is in the list exactly 3 times.
    if idents.count(thread.get_ident()) != 3:
        raise test_support.TestFailed, \
              "Couldn't find main thread correctly in the list"

# Tests which use _testcapi helpers
class OtherTests(unittest.TestCase):
    def test_exc_L(self):
        # This used to raise a SystemError(bad internal call)
        self.assertRaises(TypeError, _testcapi.getargs_L, "String")

try:
    _testcapi._test_thread_state
    have_thread_state = True
except AttributeError:
    have_thread_state = False

if have_thread_state:
    TestThreadState()
    import threading
    t=threading.Thread(target=TestThreadState)
    t.start()

def test_main():
    test_support.run_unittest(OtherTests)

if __name__=='__main__':
    test_main()