Differential D3070 Diff 10075 tests/python/collada/mesh/test_mesh_simple.py

Changeset View

Standalone View

tests/python/collada/mesh/test_mesh_simple.py

This file was added.

				#!/usr/bin/env python3
				# ##### BEGIN GPL LICENSE BLOCK #####
				#
				# This program is free software; you can redistribute it and/or
				# modify it under the terms of the GNU General Public License
				# as published by the Free Software Foundation; either version 2
				# of the License, or (at your option) any later version.
				#
				# This program is distributed in the hope that it will be useful,
				# but WITHOUT ANY WARRANTY; without even the implied warranty of
				# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
				# GNU General Public License for more details.
				#
				# You should have received a copy of the GNU General Public License
				# along with this program; if not, write to the Free Software Foundation,
				# Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
				#
				# ##### END GPL LICENSE BLOCK #####
				#
				# Call as follows:
				# python collada_mesh_simple.py --blender PATH_TO_BLENDER_EXE --testdir PATH_TO_SVN/lib/tests/collada/mesh
				#
				import sys
				import bpy
				import argparse
				import functools
				import shutil
				import tempfile
				import unittest
				import difflib
				import pathlib
				from pathlib import Path

				def with_tempdir(wrapped):
				"""Creates a temporary directory for the function, cleaning up after it returns normally.

				When the wrapped function raises an exception, the contents of the temporary directory
				remain available for manual inspection.

				The wrapped function is called with an extra positional argument containing
				the pathlib.Path() of the temporary directory.
				"""

				@functools.wraps(wrapped)
				def decorator(args, *kwargs):
				dirname = tempfile.mkdtemp(prefix='blender-collada-test')
				#print("Using tempdir %s" % dirname)
				try:
				retval = wrapped(args, pathlib.Path(dirname), *kwargs)
				except:
				print('Exception in %s, not cleaning up temporary directory %s' % (wrapped, dirname))
				raise
				else:
				shutil.rmtree(dirname)
				return retval

				return decorator

				class AbstractColladaTest(unittest.TestCase):

				@classmethod
				def setUpClass(cls):
				cls.testdir = pathlib.Path(args.testdir)

				def checkdae(self, reference, export):
				"""
				collada verifier checks if exported dae file is the same as reference dae
				"""

				ref = open(reference)
				exp = open(export)
				diff=difflib.unified_diff(ref.readlines(), exp.readlines(), lineterm='', n=0)
				ref.close()
				exp.close()

				diff_count = 0;
				for line in diff:
				error = True
				for prefix in ('---', '+++', '@@'):
				# Ignore diff metadata
				if line.startswith(prefix):
				error=False
				break
				else:
				# Ignore time stamps
				for ignore in ('<created>', '<modified>', '<authoring_tool>'):
				if line[1:].strip().startswith(ignore):
				error=False
				break
				if error:
				diff_count +=1
				print ("%s"%line.strip())

				if diff_count > 0:
				print("Generated file differs from reference")
				print("reference: %s" % reference)
				print("result : %s" % export)

				return diff_count == 0

				class MeshExportTest(AbstractColladaTest):
				@with_tempdir
				def test_export_single_mesh(self, tempdir: pathlib.Path):
				test = "mesh_simple_001"
				reference_dae = self.testdir / Path("%s.dae" % test)
				outfile = tempdir / Path("%s_out.dae" % test)

				bpy.ops.wm.collada_export(filepath="%s" % str(outfile),
				check_existing=True,
				filemode=8,
				display_type="DEFAULT",
				sort_method="FILE_SORT_ALPHA",
				apply_modifiers=False,
				export_mesh_type=0,
				export_mesh_type_selection="view",
				selected=False,
				include_children=False,
				include_armatures=False,
				include_shapekeys=True,
				deform_bones_only=False,
				sampling_rate=0,
				active_uv_only=False,
				use_texture_copies=True,
				triangulate=False,
				use_object_instantiation=True,
				use_blender_profile=True,
				sort_by_name=False,
				export_transformation_type=0,
				export_transformation_type_selection="matrix",
				export_texture_type=0,
				export_texture_type_selection="mat",
				open_sim=False,
				limit_precision=False,
				keep_bind_info=False)

				# Now check the resulting Collada file.
				self.assertTrue(self.checkdae(reference_dae, outfile))

				if __name__ == '__main__':
				sys.argv = [__file__] + (sys.argv[sys.argv.index("--") + 1:] if "--" in sys.argv else [])
				parser = argparse.ArgumentParser()
				parser.add_argument('--testdir', required=True)
				args, remaining = parser.parse_known_args()
				unittest.main(argv=sys.argv[0:1]+remaining)
				No newline at end of file