Rewrite test_cprofile as unittest (and rename the file to be consistent

with all other test files). Written for GHOP by Benjamin Peterson.

1 files changed, 208 insertions, 0 deletions

diff --git a/Lib/test/test_cprofile.py b/Lib/test/test_cprofile.py
new file mode 100644
index 0000000..fdbd542
--- /dev/null
+++ b/Lib/test/test_cprofile.py
@@ -0,0 +1,208 @@
+import sys
+import cProfile
+import pstats
+import test.test_support
+
+#################################
+#       Warning!
+# This stuff is touchy. If you modify anything above the test_main function,
+# you'll have to regenerate the stats for the doctest!
+################################
+
+TICKS = 42000
+
+def timer():
+    return TICKS
+
+def testfunc():
+    # 1 call
+    # 1000 ticks total: 270 ticks local, 730 ticks in subfunctions
+    global TICKS
+    TICKS += 99
+    helper()                            # 300
+    helper()                            # 300
+    TICKS += 171
+    factorial(14)                       # 130
+
+def factorial(n):
+    # 23 calls total
+    # 170 ticks total, 150 ticks local
+    # 3 primitive calls, 130, 20 and 20 ticks total
+    # including 116, 17, 17 ticks local
+    global TICKS
+    if n > 0:
+        TICKS += n
+        return mul(n, factorial(n-1))
+    else:
+        TICKS += 11
+        return 1
+
+def mul(a, b):
+    # 20 calls
+    # 1 tick, local
+    global TICKS
+    TICKS += 1
+    return a * b
+
+def helper():
+    # 2 calls
+    # 300 ticks total: 20 ticks local, 260 ticks in subfunctions
+    global TICKS
+    TICKS += 1
+    helper1()                           # 30
+    TICKS += 2
+    helper1()                           # 30
+    TICKS += 6
+    helper2()                           # 50
+    TICKS += 3
+    helper2()                           # 50
+    TICKS += 2
+    helper2()                           # 50
+    TICKS += 5
+    helper2_indirect()                  # 70
+    TICKS += 1
+
+def helper1():
+    # 4 calls
+    # 30 ticks total: 29 ticks local, 1 tick in subfunctions
+    global TICKS
+    TICKS += 10
+    hasattr(C(), "foo")                 # 1
+    TICKS += 19
+    lst = []
+    lst.append(42)                      # 0
+    sys.exc_info()                      # 0
+
+def helper2_indirect():
+    helper2()                           # 50
+    factorial(3)                        # 20
+
+def helper2():
+    # 8 calls
+    # 50 ticks local: 39 ticks local, 11 ticks in subfunctions
+    global TICKS
+    TICKS += 11
+    hasattr(C(), "bar")                 # 1
+    TICKS += 13
+    subhelper()                         # 10
+    TICKS += 15
+
+def subhelper():
+    # 8 calls
+    # 10 ticks total: 8 ticks local, 2 ticks in subfunctions
+    global TICKS
+    TICKS += 2
+    for i in range(2):                  # 0
+        try:
+            C().foo                     # 1 x 2
+        except AttributeError:
+            TICKS += 3                  # 3 x 2
+
+class C:
+    def __getattr__(self, name):
+        # 28 calls
+        # 1 tick, local
+        global TICKS
+        TICKS += 1
+        raise AttributeError
+
+def test_main():
+    """
+    >>> prof = cProfile.Profile(timer, 0.001)
+    >>> prof.runctx("testfunc()", globals(), locals()) #doctest: +ELLIPSIS
+    <cProfile.Profile object at ...>
+    >>> timer()
+    43000
+    >>> stats = pstats.Stats(prof)
+    >>> stats.strip_dirs().sort_stats("stdname") #doctest: +ELLIPSIS
+    <pstats.Stats ...>
+    >>> stats.print_stats() #doctest: +ELLIPSIS
+             126 function calls (106 primitive calls) in 1.000 CPU seconds
+    <BLANKLINE>
+       Ordered by: standard name
+    <BLANKLINE>
+       ncalls  tottime  percall  cumtime  percall filename:lineno(function)
+            1    0.000    0.000    1.000    1.000 <string>:1(<module>)
+           28    0.028    0.001    0.028    0.001 test_cprofile.py:102(__getattr__)
+            1    0.270    0.270    1.000    1.000 test_cprofile.py:17(testfunc)
+         23/3    0.150    0.007    0.170    0.057 test_cprofile.py:27(factorial)
+           20    0.020    0.001    0.020    0.001 test_cprofile.py:40(mul)
+            2    0.040    0.020    0.600    0.300 test_cprofile.py:47(helper)
+            4    0.116    0.029    0.120    0.030 test_cprofile.py:65(helper1)
+            2    0.000    0.000    0.140    0.070 test_cprofile.py:76(helper2_indirect)
+            8    0.312    0.039    0.400    0.050 test_cprofile.py:80(helper2)
+            8    0.064    0.008    0.080    0.010 test_cprofile.py:90(subhelper)
+           12    0.000    0.000    0.012    0.001 {hasattr}
+            4    0.000    0.000    0.000    0.000 {method 'append' of 'list' objects}
+            1    0.000    0.000    0.000    0.000 {method 'disable' of '_lsprof.Profiler' objects}
+            8    0.000    0.000    0.000    0.000 {range}
+            4    0.000    0.000    0.000    0.000 {sys.exc_info}
+    <BLANKLINE>
+    <BLANKLINE>
+    <pstats.Stats instance at ...>
+    >>> stats.print_callers() #doctest: +ELLIPSIS
+       Ordered by: standard name
+    <BLANKLINE>
+    Function                                          was called by...
+                                                          ncalls  tottime  cumtime
+    <string>:1(<module>)                              <-
+    test_cprofile.py:102(__getattr__)                 <-      16    0.016    0.016  test_cprofile.py:90(subhelper)
+                                                              12    0.012    0.012  {hasattr}
+    test_cprofile.py:17(testfunc)                     <-       1    0.270    1.000  <string>:1(<module>)
+    test_cprofile.py:27(factorial)                    <-       1    0.014    0.130  test_cprofile.py:17(testfunc)
+                                                            20/3    0.130    0.147  test_cprofile.py:27(factorial)
+                                                               2    0.006    0.040  test_cprofile.py:76(helper2_indirect)
+    test_cprofile.py:40(mul)                          <-      20    0.020    0.020  test_cprofile.py:27(factorial)
+    test_cprofile.py:47(helper)                       <-       2    0.040    0.600  test_cprofile.py:17(testfunc)
+    test_cprofile.py:65(helper1)                      <-       4    0.116    0.120  test_cprofile.py:47(helper)
+    test_cprofile.py:76(helper2_indirect)             <-       2    0.000    0.140  test_cprofile.py:47(helper)
+    test_cprofile.py:80(helper2)                      <-       6    0.234    0.300  test_cprofile.py:47(helper)
+                                                               2    0.078    0.100  test_cprofile.py:76(helper2_indirect)
+    test_cprofile.py:90(subhelper)                    <-       8    0.064    0.080  test_cprofile.py:80(helper2)
+    {hasattr}                                         <-       4    0.000    0.004  test_cprofile.py:65(helper1)
+                                                               8    0.000    0.008  test_cprofile.py:80(helper2)
+    {method 'append' of 'list' objects}               <-       4    0.000    0.000  test_cprofile.py:65(helper1)
+    {method 'disable' of '_lsprof.Profiler' objects}  <-
+    {range}                                           <-       8    0.000    0.000  test_cprofile.py:90(subhelper)
+    {sys.exc_info}                                    <-       4    0.000    0.000  test_cprofile.py:65(helper1)
+    <BLANKLINE>
+    <BLANKLINE>
+    <pstats.Stats instance at ...>
+    >>> stats.print_callees() #doctest: +ELLIPSIS
+       Ordered by: standard name
+    <BLANKLINE>
+    Function                                          called...
+                                                          ncalls  tottime  cumtime
+    <string>:1(<module>)                              ->       1    0.270    1.000  test_cprofile.py:17(testfunc)
+    test_cprofile.py:102(__getattr__)                 ->
+    test_cprofile.py:17(testfunc)                     ->       1    0.014    0.130  test_cprofile.py:27(factorial)
+                                                               2    0.040    0.600  test_cprofile.py:47(helper)
+    test_cprofile.py:27(factorial)                    ->    20/3    0.130    0.147  test_cprofile.py:27(factorial)
+                                                              20    0.020    0.020  test_cprofile.py:40(mul)
+    test_cprofile.py:40(mul)                          ->
+    test_cprofile.py:47(helper)                       ->       4    0.116    0.120  test_cprofile.py:65(helper1)
+                                                               2    0.000    0.140  test_cprofile.py:76(helper2_indirect)
+                                                               6    0.234    0.300  test_cprofile.py:80(helper2)
+    test_cprofile.py:65(helper1)                      ->       4    0.000    0.004  {hasattr}
+                                                               4    0.000    0.000  {method 'append' of 'list' objects}
+                                                               4    0.000    0.000  {sys.exc_info}
+    test_cprofile.py:76(helper2_indirect)             ->       2    0.006    0.040  test_cprofile.py:27(factorial)
+                                                               2    0.078    0.100  test_cprofile.py:80(helper2)
+    test_cprofile.py:80(helper2)                      ->       8    0.064    0.080  test_cprofile.py:90(subhelper)
+                                                               8    0.000    0.008  {hasattr}
+    test_cprofile.py:90(subhelper)                    ->      16    0.016    0.016  test_cprofile.py:102(__getattr__)
+                                                               8    0.000    0.000  {range}
+    {hasattr}                                         ->      12    0.012    0.012  test_cprofile.py:102(__getattr__)
+    {method 'append' of 'list' objects}               ->
+    {method 'disable' of '_lsprof.Profiler' objects}  ->
+    {range}                                           ->
+    {sys.exc_info}                                    ->
+    <BLANKLINE>
+    <BLANKLINE>
+    <pstats.Stats instance at ...>
+    """
+    from test import test_cprofile
+    test.test_support.run_doctest(test_cprofile)
+
+if __name__ == "__main__":
+    test_main()