blob: 87be9ee7755f3533249b389ddcd4709fc153555c [file] [log] [blame]
Tim Petersd66595f2001-02-04 03:09:53 +00001# Run the _testcapi module tests (tests for the Python/C API): by defn,
Guido van Rossum361c5352001-04-13 17:03:04 +00002# these are all functions _testcapi exports whose name begins with 'test_'.
Tim Peters9ea17ac2001-02-02 05:57:15 +00003
Martin v. Löwis6ce7ed22005-03-03 12:26:35 +00004import sys
Barry Warsaw04f357c2002-07-23 19:04:11 +00005from test import test_support
Tim Petersd66595f2001-02-04 03:09:53 +00006import _testcapi
Tim Peters9ea17ac2001-02-02 05:57:15 +00007
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00008def test_main():
9
10 for name in dir(_testcapi):
11 if name.startswith('test_'):
12 test = getattr(_testcapi, name)
13 if test_support.verbose:
Guido van Rossumbe19ed72007-02-09 05:37:30 +000014 print("internal", name)
Collin Winter3add4d72007-08-29 23:37:32 +000015 test()
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000016
17 # some extra thread-state tests driven via _testcapi
18 def TestThreadState():
Tim Peters9ea17ac2001-02-02 05:57:15 +000019 if test_support.verbose:
Guido van Rossumbe19ed72007-02-09 05:37:30 +000020 print("auto-thread-state")
Mark Hammond8d98d2c2003-04-19 15:41:53 +000021
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000022 idents = []
Mark Hammond8d98d2c2003-04-19 15:41:53 +000023
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000024 def callback():
25 idents.append(thread.get_ident())
Mark Hammond8d98d2c2003-04-19 15:41:53 +000026
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000027 _testcapi._test_thread_state(callback)
28 a = b = callback
29 time.sleep(1)
30 # Check our main thread is in the list exactly 3 times.
31 if idents.count(thread.get_ident()) != 3:
Collin Winter3add4d72007-08-29 23:37:32 +000032 raise test_support.TestFailed(
33 "Couldn't find main thread correctly in the list")
Mark Hammond8d98d2c2003-04-19 15:41:53 +000034
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000035 try:
36 _testcapi._test_thread_state
37 have_thread_state = True
38 except AttributeError:
39 have_thread_state = False
Tim Peters0eadaac2003-04-24 16:02:54 +000040
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000041 if have_thread_state:
Christian Heimes5e696852008-04-09 08:37:03 +000042 import thread
43 import time
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000044 TestThreadState()
45 import threading
46 t=threading.Thread(target=TestThreadState)
47 t.start()
48 t.join()
Mark Hammond8d98d2c2003-04-19 15:41:53 +000049
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000050if __name__ == "__main__":
51 test_main()