1 | #!/usr/bin/env python2
|
2 | """
|
3 | files.py
|
4 | """
|
5 | from __future__ import print_function
|
6 |
|
7 | import os
|
8 | import sys
|
9 |
|
10 | from mycpp import mylib
|
11 | from mycpp.mylib import log
|
12 |
|
13 |
|
14 | def run_tests():
|
15 | # type: () -> None
|
16 |
|
17 | f = mylib.BufWriter()
|
18 | for i in xrange(30):
|
19 | f.write(chr(i + 65))
|
20 |
|
21 | contents = f.getvalue()
|
22 | log('Wrote %d bytes to StringIO', len(contents))
|
23 |
|
24 | # the f(g(), h()) problem with mark and sweep!!!
|
25 | if 1:
|
26 | log('contents = %s ... %s', contents[:10], contents[-10:])
|
27 |
|
28 | # Does not have the problem
|
29 | if 0:
|
30 | a = contents[:10]
|
31 | b = contents[-10:]
|
32 | log('contents = %s ... %s', a, b)
|
33 |
|
34 | f2 = mylib.Stdout()
|
35 | f2.write('stdout\n')
|
36 |
|
37 |
|
38 | def run_benchmarks():
|
39 | # type: () -> None
|
40 | n = 10000
|
41 |
|
42 | result = 0
|
43 |
|
44 | i = 0
|
45 | while i < n:
|
46 | f = mylib.BufWriter()
|
47 | for j in xrange(30):
|
48 | f.write(chr(j + 65))
|
49 |
|
50 | result += len(f.getvalue())
|
51 |
|
52 | mylib.MaybeCollect()
|
53 |
|
54 | i += 1
|
55 | log('Ran %d iterations', n)
|
56 | log('result = %d', result)
|
57 |
|
58 |
|
59 | if __name__ == '__main__':
|
60 | if os.getenv('BENCHMARK'):
|
61 | log('Benchmarking...')
|
62 | run_benchmarks()
|
63 | else:
|
64 | run_tests()
|