test_capi.py 1.25 KB
Newer Older
1
# Run the _testcapi module tests (tests for the Python/C API):  by defn,
2
# these are all functions _testcapi exports whose name begins with 'test_'.
3

4
import sys
5
from test import test_support
6
import _testcapi
7

8
for name in dir(_testcapi):
9
    if name.startswith('test_'):
10
        test = getattr(_testcapi, name)
11 12 13 14
        if test_support.verbose:
            print "internal", name
        try:
            test()
15
        except _testcapi.error:
16
            raise test_support.TestFailed, sys.exc_info()[1]
17 18 19 20 21 22 23 24 25 26 27 28 29

# 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())
Tim Peters's avatar
Tim Peters committed
30

31 32 33 34 35 36 37 38 39 40 41 42
    _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"

try:
    _testcapi._test_thread_state
    have_thread_state = True
except AttributeError:
    have_thread_state = False
Tim Peters's avatar
Tim Peters committed
43

44 45
if have_thread_state:
    TestThreadState()
46 47 48
    import threading
    t=threading.Thread(target=TestThreadState)
    t.start()