test_commands_merge_exr.py
c9234a2c13a7T53099-gpu-rendering

test_commands_merge_exr.py
View Options

			from pathlib import Path

			from test_runner import AbstractCommandTest


			class MergeProgressiveRendersCommandTest(AbstractCommandTest):
			def setUp(self):
			super().setUp()

			from flamenco_worker.commands import MergeProgressiveRendersCommand
			import tempfile

			self.tmpdir = tempfile.TemporaryDirectory()
			self.mypath = Path(__file__).parent

			self.cmd = MergeProgressiveRendersCommand(
			worker=self.fworker,
			task_id='12345',
			command_idx=0,
			)

			def tearDown(self):
			super().tearDown()
			self.tmpdir.cleanup()

			def test_happy_flow(self):
			output = Path(self.tmpdir.name) / 'merged.exr'

			settings = {
			'blender_cmd': self.find_blender_cmd(),
			'input1': str(self.mypath / 'Corn field-1k.exr'),
			'input2': str(self.mypath / 'Deventer-1k.exr'),
			'weight1': 20,
			'weight2': 100,
			'output': str(output)
			}

			task = self.cmd.run(settings)
			ok = self.loop.run_until_complete(task)
			self.assertTrue(ok)

			# Assuming that if the files exist, the merge was ok.
			self.assertTrue(output.exists())
			self.assertTrue(output.is_file())

			# Sybren disabled preview generation since we don't use those any more
			# in the studio.
			# self.assertTrue(output.with_suffix('.jpg').exists())
			# self.assertTrue(output.with_suffix('.jpg').is_file())