summaryrefslogtreecommitdiff
path: root/Examples/python/import_packages/namespace_pkg/nstest.py
blob: 9b075ad8ca898f1b2d08e7221de35d89d7ab7084 (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
import sys
import os
import shutil
import zipfile


def copyMods():
    dirs = ['path1', 'path2', 'path3']

    # Clean out any old package paths
    for d in dirs:
        if os.path.isdir(d):
            shutil.rmtree(d)

    for d in dirs:
        os.mkdir(d)
        os.mkdir(os.path.join(d, 'brave'))

    shutil.copy('robin.py', os.path.join('path1', 'brave'))
    os.system('cp _robin.* ' + os.path.join('path1', 'brave'))

    shutil.copy('robin.py', os.path.join('path2', 'brave'))
    os.system('cp _robin.* ' + os.path.join('path3', 'brave'))

    mkzip()

def mkzip():
    zf = zipfile.ZipFile("path4.zip", "w")
    zf.writestr("brave/", b'')
    zf.write('robin.py', 'brave/robin.py')
    zf.close()


def main():
    copyMods()

    # Run each test with a separate interpreter
    os.system(sys.executable + " nonpkg.py")
    os.system(sys.executable + " normal.py")
    os.system(sys.executable + " split.py")
    os.system(sys.executable + " zipsplit.py")


if __name__ == "__main__":
    main()