Differential D7489 Diff 34500 tests/python/transform_operators.py

Changeset View

Standalone View

tests/python/transform_operators.py

This file was added.

				# ##### 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 #####

				# <pep8 compliant>

				# To run all tests, use
				# blender path/to/transform_regression.blend --python
				# path/to/transform_operators.py --enable-event-simulate -- --run-all-tests

				# To run one test, use
				# blender path/to/transform_regression.blend --python
				# path/to/transform_operators.py --enable-event-simulate -- --run-test <index>
				# where <index> is the index of the test specified in the list tests.
				sybrenUnsubmitted Done Inline Actions I've never seen this before, and I feel it's also a bit clunky. Most test frameworks use the name of the test, and not its index. sybren: I've never seen this before, and I feel it's also a bit clunky. Most test frameworks use the…
				mano-wiiAuthorUnsubmitted Done Inline Actions This index is for internal use. The test name is `"transform"`. This is the same as seen in `modifier.py` which has an API similar. Basically `transform_operators.py` is a modified copy of `modifier.py` and `transform_test.py` is a modified copy of `mesh_test.py`. mano-wii: This index is for internal use. The test name is `"transform"`. This is the same as seen in…
				sybrenUnsubmitted Done Inline Actions If there is the need to manually add comments to a list to indicate the index of each item, that's an indication of a bad design. There is also a comment that explains how to use that index to run a single test, so it's not for internal use either. `modifiers.py` identifies tests by name, and not by index, so I don't really see your point. sybren: If there is the need to manually add comments to a list to indicate the index of each item…
				mano-wiiAuthorUnsubmitted Done Inline Actions I may be seeing things, but I'm sure it used to be index. I don't even know what the names of the modifier tests are. In that case it may be important to add a --help command to see these names. mano-wii: I may be seeing things, but I'm sure it used to be index. I don't even know what the names of…
				import bpy
				import os
				import sys
				import math

				sys.path.append(os.path.dirname(os.path.realpath(__file__)))
				from modules.transform_test import TestContext

				def test_translate_global_objectmode(test_ctx):
				obj_test = bpy.data.objects['MyObject']
				obj_expected = bpy.data.objects['MyObject_trans_global_object_expected']
				obj_dupl = test_ctx.duplicate_object(obj_test)
				test_ctx.select_objects((obj_dupl,), active=obj_dupl)
				bpy.ops.transform.translate(test_ctx.override_context,
				value=(0,0,1),
				constraint_axis=(True,True,True),)
				test_ctx.compare_matrix(obj_dupl, obj_expected, test_translate_global_objectmode.__name__)

				def test_translate_global_editmode_a(test_ctx):
				obj_test = bpy.data.objects['MyObject']
				sybrenUnsubmitted Done Inline Actions What are "edit mode a" and "b"? What is being tested here? The way the test code is written, it's hard to see the difference between the "a" and "b" functions. sybren: What are "edit mode a" and "b"? What is being tested here? The way the test code is written…
				obj_expected = bpy.data.objects['MyObject_trans_global_edit_expected']
				obj_dupl = test_ctx.duplicate_object(obj_test)
				test_ctx.select_objects((obj_dupl,), active=obj_dupl)
				test_ctx.mode_set('EDIT')
				bpy.ops.mesh.select_all(test_ctx.override_context, action="SELECT")
				bpy.ops.transform.translate(test_ctx.override_context,
				value=(0,0,1),
				constraint_axis=(True,True,True),)
				test_ctx.compare_data(obj_dupl, obj_expected, test_translate_global_editmode_a.__name__)

				def test_translate_global_editmode_b(test_ctx):
				obj_test = bpy.data.objects['MyObject']
				obj_expected = bpy.data.objects['MyObject_trans_global_edit_expected']
				obj_dupl = test_ctx.duplicate_object(obj_test)
				test_ctx.select_objects((obj_dupl,), active=obj_dupl)
				test_ctx.mode_set('EDIT')
				bpy.ops.mesh.select_all(test_ctx.override_context, action="SELECT")
				bpy.ops.transform.translate(test_ctx.override_context,
				value=(0,0,1),
				constraint_axis=(True,True,True),
				orient_type='GLOBAL',)
				test_ctx.compare_data(obj_dupl, obj_expected, test_translate_global_editmode_b.__name__)

				def test_translate_local_objectmode(test_ctx):
				obj_test = bpy.data.objects['MyObject_transformed']
				obj_expected = bpy.data.objects['MyObject_trans_local_object_expected']
				obj_dupl = test_ctx.duplicate_object(obj_test)
				test_ctx.select_objects((obj_dupl,), active=obj_dupl)
				bpy.context.scene.transform_orientation_slots[0].type = 'LOCAL'
				bpy.ops.transform.translate(test_ctx.override_context,
				value=(0,0,1),
				constraint_axis=(True,True,True),)
				test_ctx.compare_matrix(obj_dupl, obj_expected, test_translate_local_objectmode.__name__)
				sergeyUnsubmitted Done Inline Actions I am sure you can make this more readable about what is the input, what is the expected expression. Having name-less lists and tuples do not help readability and maintainability. sergey: I am sure you can make this more readable about what is the input, what is the expected…

				def test_translate_local_editmode_a(test_ctx):
				obj_test = bpy.data.objects['MyObject_transformed']
				obj_expected = bpy.data.objects['MyObject_trans_local_edit_expected']
				obj_dupl = test_ctx.duplicate_object(obj_test)
				test_ctx.select_objects((obj_dupl,), active=obj_dupl)
				test_ctx.mode_set('EDIT')
				bpy.context.scene.transform_orientation_slots[0].type = 'LOCAL'
				bpy.ops.mesh.select_all(test_ctx.override_context, action="SELECT")
				bpy.ops.transform.translate(test_ctx.override_context,
				value=(0,0,1),
				constraint_axis=(True,True,True),)
				test_ctx.compare_data(obj_dupl, obj_expected, test_translate_local_editmode_a.__name__)

				def test_translate_local_editmode_b(test_ctx):
				obj_test = bpy.data.objects['MyObject_transformed']
				obj_expected = bpy.data.objects['MyObject_trans_local_edit_expected']
				obj_dupl = test_ctx.duplicate_object(obj_test)
				test_ctx.select_objects((obj_dupl,), active=obj_dupl)
				test_ctx.mode_set('EDIT')
				bpy.context.scene.transform_orientation_slots[0].type = 'LOCAL'
				bpy.ops.mesh.select_all(test_ctx.override_context, action="SELECT")
				bpy.ops.transform.translate(test_ctx.override_context,
				value=(0,0,1),
				constraint_axis=(True,True,True),
				orient_type='LOCAL',)
				test_ctx.compare_data(obj_dupl, obj_expected, test_translate_local_editmode_b.__name__)

				def test_rotate_local_objectmode_a(test_ctx):
				obj_test = bpy.data.objects['MyObject_transformed']
				obj_expected = bpy.data.objects['MyObject_rot_local_object_expected']
				obj_dupl = test_ctx.duplicate_object(obj_test)
				test_ctx.select_objects((obj_dupl,), active=obj_dupl)
				bpy.context.scene.transform_orientation_slots[0].type = 'LOCAL'
				bpy.ops.transform.rotate(test_ctx.override_context,
				value=math.radians(90),
				constraint_axis=(False,False,True),)
				test_ctx.compare_matrix(obj_dupl, obj_expected, test_rotate_local_objectmode_a.__name__)

				def test_rotate_local_objectmode_b(test_ctx):
				obj_test = bpy.data.objects['MyObject_transformed']
				obj_expected = bpy.data.objects['MyObject_rot_local_object_expected']
				obj_dupl = test_ctx.duplicate_object(obj_test)
				test_ctx.select_objects((obj_dupl,), active=obj_dupl)
				bpy.context.scene.transform_orientation_slots[0].type = 'LOCAL'
				bpy.ops.transform.rotate(test_ctx.override_context,
				value=math.radians(90),
				constraint_axis=(False,False,True),
				orient_type='LOCAL',)
				test_ctx.compare_matrix(obj_dupl, obj_expected, test_rotate_local_objectmode_b.__name__)

				def test_rotate_view_objectmode_a(test_ctx):
				obj_test = bpy.data.objects['MyObject_transformed']
				obj_expected = bpy.data.objects['MyObject_rot_view_object_expected_0']
				obj_dupl = test_ctx.duplicate_object(obj_test)
				test_ctx.select_objects((obj_dupl,), active=obj_dupl)
				bpy.context.scene.transform_orientation_slots[0].type = 'VIEW'
				bpy.ops.transform.rotate(test_ctx.override_context,
				value=math.radians(90),)
				test_ctx.compare_matrix(obj_dupl, obj_expected, test_rotate_view_objectmode_a.__name__)

				def test_rotate_view_objectmode_b(test_ctx):
				obj_test = bpy.data.objects['MyObject']
				obj_expected = bpy.data.objects['MyObject_rot_view_object_expected_1']
				obj_dupl = test_ctx.duplicate_object(obj_test)
				test_ctx.select_objects((obj_dupl,), active=obj_dupl)
				bpy.context.scene.transform_orientation_slots[0].type = 'VIEW'
				bpy.ops.transform.rotate(test_ctx.override_context,
				value=math.radians(90),
				constraint_axis=(False,False,True),
				orient_type='VIEW',
				orient_matrix_type='VIEW',
				orient_matrix=((1.0,0.0,0.0),
				(0.0,0.0,1.0),
				(0.0,-1.0,0.0)), # -Y View
				)
				test_ctx.compare_matrix(obj_dupl, obj_expected, test_rotate_view_objectmode_b.__name__)

				def test_rotate_local_weightpaintmode(test_ctx):
				obj_arm = bpy.data.objects['Armature']
				obj_test = bpy.data.objects['MyObject_rigged']
				obj_expected = bpy.data.objects['MyObject_rot_local_weightpaint_expected']
				test_ctx.select_objects((obj_arm, obj_test), active=obj_test)
				obj_dupl_act = test_ctx.duplicate_selected()
				test_ctx.mode_set('WEIGHT_PAINT')
				bpy.context.scene.transform_orientation_slots[0].type = 'LOCAL'
				bpy.ops.transform.rotate(test_ctx.override_context,
				value=math.radians(90),
				constraint_axis=(True,False,False),
				)
				test_ctx.compare_data(obj_dupl_act, obj_expected, test_rotate_local_weightpaintmode.__name__)

				def test_rotate_normal_editmode_individual_orig_a(test_ctx):
				scene = bpy.context.scene
				obj_test = bpy.data.objects['MyObject_transformed']
				obj_expected = bpy.data.objects['MyObject_rot_normal_edit_individual_orig_expected']
				obj_dupl = test_ctx.duplicate_object(obj_test)
				test_ctx.select_objects((obj_dupl,), active=obj_dupl)
				test_ctx.mode_set('EDIT')
				scene.transform_orientation_slots[0].type = 'NORMAL'
				scene.tool_settings.transform_pivot_point = 'INDIVIDUAL_ORIGINS'
				bpy.ops.mesh.select_all(test_ctx.override_context, action="SELECT")
				bpy.ops.transform.rotate(test_ctx.override_context,
				value=math.radians(45),
				constraint_axis=(False,False,True),
				)
				test_ctx.compare_data(obj_dupl, obj_expected, test_rotate_normal_editmode_individual_orig_a.__name__)

				def test_rotate_normal_editmode_individual_orig_b(test_ctx):
				obj_test = bpy.data.objects['MyObject_transformed']
				obj_expected = bpy.data.objects['MyObject_rot_normal_edit_individual_orig_expected']
				obj_dupl = test_ctx.duplicate_object(obj_test)
				test_ctx.select_objects((obj_dupl,), active=obj_dupl)
				test_ctx.mode_set('EDIT')
				scene = bpy.context.scene
				scene.transform_orientation_slots[0].type = 'NORMAL'
				scene.tool_settings.transform_pivot_point = 'INDIVIDUAL_ORIGINS'
				bpy.ops.mesh.select_all(test_ctx.override_context, action="SELECT")
				bpy.ops.transform.rotate(test_ctx.override_context,
				value=math.radians(45),
				constraint_axis=(False,False,True),
				orient_type='NORMAL',)
				test_ctx.compare_data(obj_dupl, obj_expected, test_rotate_normal_editmode_individual_orig_b.__name__)

				def test_rotate_normal_editmode_individual_orig_c(test_ctx):
				obj_test = bpy.data.objects['MyObject_transformed']
				obj_expected = bpy.data.objects['MyObject_rot_normal_edit_individual_orig_expected']
				obj_dupl = test_ctx.duplicate_object(obj_test)
				test_ctx.select_objects((obj_dupl,), active=obj_dupl)
				test_ctx.mode_set('EDIT')
				scene = bpy.context.scene
				scene.transform_orientation_slots[0].type = 'NORMAL'
				scene.tool_settings.transform_pivot_point = 'INDIVIDUAL_ORIGINS'
				bpy.ops.mesh.select_all(test_ctx.override_context, action="SELECT")
				bpy.ops.transform.rotate(test_ctx.override_context,
				value=math.radians(45),
				constraint_axis=(False,False,True),
				orient_type='NORMAL',
				orient_matrix_type='NORMAL',
				orient_matrix=((1,0,0),
				(0,1,0),
				(0,0,1)),)
				test_ctx.compare_data(obj_dupl, obj_expected, test_rotate_normal_editmode_individual_orig_c.__name__)

				def test_translate_w_events_gg_editmode(test_ctx):
				import bmesh
				obj_test = bpy.data.objects['MyObject_transformed']
				obj_expected = bpy.data.objects['MyObject_trans_event_gg_expected']
				obj_dupl = test_ctx.duplicate_object(obj_test)
				test_ctx.select_objects((obj_dupl,), active=obj_dupl)
				test_ctx.mode_set('EDIT')
				bpy.ops.mesh.select_all(test_ctx.override_context, action="DESELECT")
				bm = bmesh.from_edit_mesh(obj_dupl.data)
				bm.faces.ensure_lookup_table()
				bm.faces[0].select_set(True)
				bpy.ops.transform.translate(test_ctx.override_context, 'INVOKE_DEFAULT',)
				window = test_ctx.override_context['window']
				window.event_simulate('Z', 'PRESS')
				window.event_simulate('Z', 'RELEASE')
				window.event_simulate('Z', 'PRESS')
				window.event_simulate('Z', 'RELEASE')
				window.event_simulate('ONE', 'PRESS', unicode="1")
				window.event_simulate('ONE', 'RELEASE')
				window.event_simulate('RET', 'PRESS')
				window.event_simulate('RET', 'RELEASE')
				test_ctx.compare_data(obj_dupl, obj_expected, test_translate_w_events_gg_editmode.__name__)

				def main():
				arg_keep_open = False
				arg_event_simulate = False
				index = None
				command = list(sys.argv)
				for i, cmd in enumerate(command):
				if cmd == "--keep-open":
				arg_keep_open = True
				elif cmd == "--run-all-tests":
				index = None
				elif cmd == "--run-test":
				index = int(command[i + 1])
				elif cmd == "--enable-event-simulate":
				assert not "--background" in command
				arg_event_simulate = True
				sybrenUnsubmitted Done Inline Actions Why isn't this using `argparse` instead? sybren: Why isn't this using `argparse` instead?

				os.environ["BLENDER_VERBOSE"] = "1"
				sybrenUnsubmitted Done Inline Actions What does this do? sybren: What does this do?
				mano-wiiAuthorUnsubmitted Done Inline Actions Removed because it was not meant to be set here. This environment variable is used in some blender test files. mano-wii: Removed because it was not meant to be set here. This environment variable is used in some…
				funcs = [\
				test_translate_global_objectmode, # 0
				test_translate_global_editmode_a, # 1
				test_translate_global_editmode_b, # 2
				test_translate_local_objectmode, # 3
				test_translate_local_editmode_a, # 4
				test_translate_local_editmode_b, # 5
				test_rotate_local_objectmode_a, # 6
				test_rotate_local_objectmode_b, # 7
				test_rotate_view_objectmode_a, # 8
				test_rotate_view_objectmode_b, # 9
				test_rotate_local_weightpaintmode, #10

				test_rotate_normal_editmode_individual_orig_a, # 11
				test_rotate_normal_editmode_individual_orig_b, # 12
				test_rotate_normal_editmode_individual_orig_c, # 13
				]

				if arg_event_simulate:
				funcs.extend([\
				test_translate_w_events_gg_editmode, # 14
				])

				if index is None:
				test_ctx = TestContext(funcs)
				else:
				test_ctx = TestContext([funcs[index]])
				test_ctx.cleanup = False

				def on_error():
				if not arg_keep_open:
				import traceback
				traceback.print_exc()
				sys.exit(1)

				def on_exit():
				if not arg_keep_open:
				#bpy.ops.wm.quit_blender()
				sys.exit(0)

				test_ctx.run(on_error, on_exit)

				if __name__ == "__main__":
				try:
				main()
				except:
				import traceback
				traceback.print_exc()
				sys.exit(1)