PEP8 and updated docu strings.

pull/416/head
Oliver Nagy 2014-11-23 15:31:35 +11:00
parent 3721b19a61
commit 1e9f3a802d
1 changed files with 16 additions and 15 deletions

View File

@ -18,35 +18,36 @@ sys.path.insert(0, '..')
import sample import sample
from pyassimp import errors from pyassimp import errors
# paths to be walkd recursively # Paths to model files.
basepaths = [os.path.join('..','..','..','test','models'), os.path.join('..','..','..','test','models-nonbsd')] basepaths = [os.path.join('..', '..', '..', 'test', 'models'),
os.path.join('..', '..', '..', 'test', 'models-nonbsd')]
# Valid extensions for 3D model files.
extensions = ['.3ds', '.x', '.lwo', '.obj', '.md5mesh', '.dxf', '.ply', '.stl',
'.dae', '.md5anim', '.lws', '.irrmesh', '.nff', '.off', '.blend']
# file extensions to be considered
extensions = ['.3ds','.x','.lwo','.obj','.md5mesh','.dxf','.ply','.stl','.dae','.md5anim','.lws','.irrmesh','.nff','.off','.blend']
def run_tests(): def run_tests():
ok,err = 0,0 ok, err = 0, 0
for path in basepaths: for path in basepaths:
print("Looking for models in %s..." % path) print("Looking for models in %s..." % path)
for root, dirs, files in os.walk(path): for root, dirs, files in os.walk(path):
for afile in files: for afile in files:
base,ext = os.path.splitext(afile) base, ext = os.path.splitext(afile)
if ext in extensions: if ext in extensions:
try: try:
sample.main(os.path.join(root,afile)) sample.main(os.path.join(root, afile))
ok += 1 ok += 1
except errors.AssimpError as error: except errors.AssimpError as error:
# assimp error is fine, this is a controlled case # Assimp error is fine; this is a controlled case.
print(error) print(error)
err += 1 err += 1
except Exception: except Exception:
print("Error encountered while loading <%s>"%os.path.join(root,afile)) print("Error encountered while loading <%s>"
print('** Loaded %s models, got controlled errors for %s files' % (ok,err)) % os.path.join(root, afile))
print('** Loaded %s models, got controlled errors for %s files'
% (ok, err))
if __name__ == '__main__': if __name__ == '__main__':
run_tests() run_tests()