- fix problem with cmake, windows 'RelWithDebInfo' target, was giving error: "ImportError: No module named _socket" because of copying wrong files over.
- move test -> tests, this name is used elsewhere in lib/tests. - change interface code not to loop on a float value (clang warning), harmless, but with extreme cases an eternal loop would still be possible though unlikely.
This commit is contained in:
Campbell Barton
195
source/tests/bl_test.py
Normal file
195
source/tests/bl_test.py
Normal file
@@ -0,0 +1,195 @@
|
||||
# ##### 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>
|
||||
|
||||
import sys
|
||||
import os
|
||||
|
||||
|
||||
# may split this out into a new file
|
||||
def replace_bpy_app_version():
|
||||
""" So MD5's are pradictable from output which uses blenders versions.
|
||||
"""
|
||||
|
||||
import bpy
|
||||
|
||||
app = bpy.app
|
||||
app_fake = type(bpy)("bpy.app")
|
||||
|
||||
for attr in dir(app):
|
||||
if not attr.startswith("_"):
|
||||
setattr(app_fake, attr, getattr(app, attr))
|
||||
|
||||
app_fake.version = 0, 0, 0
|
||||
app_fake.version_string = "0.00 (sub 0)"
|
||||
bpy.app = app_fake
|
||||
|
||||
|
||||
def clear_startup_blend():
|
||||
import bpy
|
||||
|
||||
for scene in bpy.data.scenes:
|
||||
for obj in scene.objects:
|
||||
scene.objects.unlink(obj)
|
||||
|
||||
|
||||
def blend_to_md5():
|
||||
import bpy
|
||||
scene = bpy.context.scene
|
||||
ROUND = 4
|
||||
|
||||
def matrix2str(matrix):
|
||||
return "".join([str(round(axis, ROUND)) for vector in matrix for axis in vector]).encode('ASCII')
|
||||
|
||||
def coords2str(seq, attr):
|
||||
return "".join([str(round(axis, ROUND)) for vertex in seq for axis in getattr(vertex, attr)]).encode('ASCII')
|
||||
|
||||
import hashlib
|
||||
|
||||
md5 = hashlib.new("md5")
|
||||
md5_update = md5.update
|
||||
|
||||
for obj in scene.objects:
|
||||
md5_update(matrix2str(obj.matrix_world))
|
||||
data = obj.data
|
||||
|
||||
if type(data) == bpy.types.Mesh:
|
||||
md5_update(coords2str(data.vertices, "co"))
|
||||
elif type(data) == bpy.types.Curve:
|
||||
for spline in data.splines:
|
||||
md5_update(coords2str(spline.bezier_points, "co"))
|
||||
md5_update(coords2str(spline.points, "co"))
|
||||
|
||||
return md5.hexdigest()
|
||||
|
||||
|
||||
def main():
|
||||
argv = sys.argv
|
||||
print(" args:", " ".join(argv))
|
||||
argv = argv[argv.index("--") + 1:]
|
||||
|
||||
def arg_extract(arg, optional=True, array=False):
|
||||
arg += "="
|
||||
if array:
|
||||
value = []
|
||||
else:
|
||||
value = None
|
||||
|
||||
i = 0
|
||||
while i < len(argv):
|
||||
if argv[i].startswith(arg):
|
||||
item = argv[i][len(arg):]
|
||||
del argv[i]
|
||||
i -= 1
|
||||
|
||||
if array:
|
||||
value.append(item)
|
||||
else:
|
||||
value = item
|
||||
break
|
||||
|
||||
i += 1
|
||||
|
||||
if (not value) and (not optional):
|
||||
print(" '%s' not set" % arg)
|
||||
sys.exit(1)
|
||||
|
||||
return value
|
||||
|
||||
run = arg_extract("--run", optional=False)
|
||||
md5 = arg_extract("--md5", optional=False)
|
||||
md5_method = arg_extract("--md5_method", optional=False) # 'SCENE' / 'FILE'
|
||||
|
||||
# only when md5_method is 'FILE'
|
||||
md5_source = arg_extract("--md5_source", optional=True, array=True)
|
||||
|
||||
# save blend file, for testing
|
||||
write_blend = arg_extract("--write-blend", optional=True)
|
||||
|
||||
# ensure files are written anew
|
||||
for f in md5_source:
|
||||
if os.path.exists(f):
|
||||
os.remove(f)
|
||||
|
||||
import bpy
|
||||
|
||||
replace_bpy_app_version()
|
||||
if not bpy.data.filepath:
|
||||
clear_startup_blend()
|
||||
|
||||
print(" Running: '%s'" % run)
|
||||
print(" MD5: '%s'!" % md5)
|
||||
|
||||
try:
|
||||
result = eval(run)
|
||||
except:
|
||||
import traceback
|
||||
traceback.print_exc()
|
||||
sys.exit(1)
|
||||
|
||||
if write_blend is not None:
|
||||
print(" Writing Blend: %s" % write_blend)
|
||||
bpy.ops.wm.save_mainfile(filepath=write_blend, check_existing=False)
|
||||
|
||||
print(" Result: '%s'" % str(result))
|
||||
if not result:
|
||||
print(" Running: %s -> False" % run)
|
||||
sys.exit(1)
|
||||
|
||||
if md5_method == 'SCENE':
|
||||
md5_new = blend_to_md5()
|
||||
elif md5_method == 'FILE':
|
||||
if not md5_source:
|
||||
print(" Missing --md5_source argument")
|
||||
sys.exit(1)
|
||||
|
||||
for f in md5_source:
|
||||
if not os.path.exists(f):
|
||||
print(" Missing --md5_source=%r argument does not point to a file")
|
||||
sys.exit(1)
|
||||
|
||||
import hashlib
|
||||
|
||||
md5_instance = hashlib.new("md5")
|
||||
md5_update = md5_instance.update
|
||||
|
||||
for f in md5_source:
|
||||
md5_update(open(f, "rb").read())
|
||||
|
||||
md5_new = md5_instance.hexdigest()
|
||||
|
||||
else:
|
||||
print(" Invalid --md5_method=%s argument is not a valid source")
|
||||
sys.exit(1)
|
||||
|
||||
if md5 != md5_new:
|
||||
print(" Running: %s\n MD5 Recieved: %s\n MD5 Expected: %s" % (run, md5_new, md5))
|
||||
sys.exit(1)
|
||||
|
||||
print(" Success: %s" % run)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
# So a python error exits(1)
|
||||
try:
|
||||
main()
|
||||
except:
|
||||
import traceback
|
||||
traceback.print_exc()
|
||||
sys.exit(1)
|
||||
Reference in New Issue
Block a user