43 lines
1.5 KiB
Python
43 lines
1.5 KiB
Python
from tests.abstract_test import AbstractBlendFileTest
|
|
|
|
from blender_asset_tracer.trace import file_sequence
|
|
|
|
|
|
class ExpandFileSequenceTest(AbstractBlendFileTest):
|
|
def setUp(self):
|
|
super().setUp()
|
|
self.imgseq = [
|
|
self.blendfiles / ("imgseq/%06d.png" % num) for num in range(210, 215)
|
|
]
|
|
|
|
def test_glob(self):
|
|
path = self.blendfiles / "imgseq/*.png"
|
|
actual = list(file_sequence.expand_sequence(path))
|
|
self.assertEqual(self.imgseq, actual)
|
|
|
|
def test_first_file(self):
|
|
path = self.blendfiles / "imgseq/000210.png"
|
|
actual = list(file_sequence.expand_sequence(path))
|
|
self.assertEqual(self.imgseq, actual)
|
|
|
|
def test_udim_sequence(self):
|
|
path = self.blendfiles / "udim/cube_UDIM.color.<UDIM>.png"
|
|
actual = list(file_sequence.expand_sequence(path))
|
|
imgseq = [
|
|
self.blendfiles / ("udim/cube_UDIM.color.%04d.png" % num) for num in range(1001, 1004)
|
|
]
|
|
self.assertEqual(imgseq, actual)
|
|
|
|
def test_nonexistent(self):
|
|
path = self.blendfiles / "nonexistant"
|
|
with self.assertRaises(file_sequence.DoesNotExist) as raises:
|
|
for result in file_sequence.expand_sequence(path):
|
|
self.fail("unexpected result %r" % result)
|
|
|
|
self.assertEqual(path, raises.exception.path)
|
|
|
|
def test_non_sequence_file(self):
|
|
path = self.blendfiles / "imgseq/LICENSE.txt"
|
|
actual = list(file_sequence.expand_sequence(path))
|
|
self.assertEqual([path], actual)
|