summaryrefslogtreecommitdiffstats
path: root/third_party/python/py/bench/localpath.py
blob: ad4fbd8e2bfd4595b584665a5cd89fff1c3c50b1 (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
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
import py
import timeit

class Listdir:
    numiter = 100000
    numentries = 100

    def setup(self):
        tmpdir = py.path.local.make_numbered_dir(self.__class__.__name__)
        for i in range(self.numentries):
            tmpdir.join(str(i))
        self.tmpdir = tmpdir

    def run(self):
        return self.tmpdir.listdir()

class Listdir_arg(Listdir):
    numiter = 100000
    numentries = 100

    def run(self):
        return self.tmpdir.listdir("47")

class Join_onearg(Listdir):
    def run(self):
        self.tmpdir.join("17")
        self.tmpdir.join("18")
        self.tmpdir.join("19")

class Join_multi(Listdir):
    def run(self):
        self.tmpdir.join("a", "b")
        self.tmpdir.join("a", "b", "c")
        self.tmpdir.join("a", "b", "c", "d")

class Check(Listdir):
    def run(self):
        self.tmpdir.check()
        self.tmpdir.check()
        self.tmpdir.check()

class CheckDir(Listdir):
    def run(self):
        self.tmpdir.check(dir=1)
        self.tmpdir.check(dir=1)
        assert not self.tmpdir.check(dir=0)

class CheckDir2(Listdir):
    def run(self):
        self.tmpdir.stat().isdir()
        self.tmpdir.stat().isdir()
        assert self.tmpdir.stat().isdir()

class CheckFile(Listdir):
    def run(self):
        self.tmpdir.check(file=1)
        assert not self.tmpdir.check(file=1)
        assert self.tmpdir.check(file=0)

if __name__ == "__main__":
    import time
    for cls in [Listdir, Listdir_arg,
                Join_onearg, Join_multi,
               Check, CheckDir, CheckDir2, CheckFile,]:

        inst = cls()
        inst.setup()
        now = time.time()
        for i in xrange(cls.numiter):
            inst.run()
        elapsed = time.time() - now
        print "%s: %d loops took %.2f seconds, per call %.6f" %(
               cls.__name__,
                cls.numiter, elapsed, elapsed / cls.numiter)