Files
test/tests/performance/tests/blend_load.py
Campbell Barton 432bfbf7a3 Cleanup: pep8
2021-07-06 12:05:27 +10:00

43 lines
918 B
Python

# Apache License, Version 2.0
import api
import os
import pathlib
def _run(filepath):
import bpy
import time
# Load once to ensure it's cached by OS
bpy.ops.wm.open_mainfile(filepath=filepath)
bpy.ops.wm.read_homefile()
# Measure loading the second time
start_time = time.time()
bpy.ops.wm.open_mainfile(filepath=filepath)
elapsed_time = time.time() - start_time
result = {'time': elapsed_time}
return result
class BlendLoadTest(api.Test):
def __init__(self, filepath):
self.filepath = filepath
def name(self):
return self.filepath.stem
def category(self):
return "blend_load"
def run(self, env, device_id):
result, _ = env.run_in_blender(_run, str(self.filepath))
return result
def generate(env):
filepaths = env.find_blend_files('*/*')
return [BlendLoadTest(filepath) for filepath in filepaths]