Frame13/ayon-core
1
0
Fork 1
mirror of https://github.com/ynput/ayon-core.git synced 2026-01-01 08:24:53 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merge remote-tracking branch 'origin/feature/909-define-basic-trait-type-using-dataclasses' into feature/909-define-basic-trait-type-using-dataclasses

Browse source
This commit is contained in:
Ondřej Samohel 2025-01-22 11:02:35 +01:00
commit 8662e42d61
No known key found for this signature in database
GPG key ID: 02376E18990A97C6
13 changed files with 414 additions and 237 deletions
Download patch file Download diff file Expand all files Collapse all files

15
tests/client/ayon_core/pipeline/traits/test_content_traits.py
View file

@ -58,7 +58,8 @@ def test_bundles() -> None:
for item in representation.get_trait(trait=Bundle).items:
sub_representation = Representation(name="test", traits=item)
assert sub_representation.contains_trait(trait=Image)
assert sub_representation.get_trait(trait=MimeType).mime_type in [
sub: MimeType = sub_representation.get_trait(trait=MimeType)
assert sub.mime_type in [
"image/jpeg", "image/tiff"
]
@ -83,7 +84,7 @@ def test_file_locations_validation() -> None:
file_paths=file_locations_list)
# this should be valid trait
file_locations_trait.validate(representation)
file_locations_trait.validate_trait(representation)
# add valid FrameRanged trait
frameranged_trait = FrameRanged(
@ -94,7 +95,7 @@ def test_file_locations_validation() -> None:
representation.add_trait(frameranged_trait)
# it should still validate fine
file_locations_trait.validate(representation)
file_locations_trait.validate_trait(representation)
# create empty file locations trait
empty_file_locations_trait = FileLocations(file_paths=[])
@ -102,7 +103,7 @@ def test_file_locations_validation() -> None:
empty_file_locations_trait
])
with pytest.raises(TraitValidationError):
empty_file_locations_trait.validate(representation)
empty_file_locations_trait.validate_trait(representation)
# create valid file locations trait but with not matching
# frame range trait
@ -118,7 +119,7 @@ def test_file_locations_validation() -> None:
representation.add_trait(invalid_sequence_trait)
with pytest.raises(TraitValidationError):
file_locations_trait.validate(representation)
file_locations_trait.validate_trait(representation)
# invalid representation with mutliple file locations but
# unrelated to either Sequence or Bundle traits
@ -164,7 +165,7 @@ def test_get_file_location_from_frame() -> None:
# test with custom regex
sequence = Sequence(
frame_padding=4,
frame_regex=r"boo_(?P<frame>\d+)\.exr")
frame_regex=r"boo_(?P<index>(?P<padding>0*)\d+)\.exr")
file_locations_list = [
FileLocation(
file_path=Path(f"/path/to/boo_{frame}.exr"),
@ -174,7 +175,7 @@ def test_get_file_location_from_frame() -> None:
for frame in range(1001, 1051)
]
file_locations_trait: FileLocations = FileLocations(
file_locations_trait = FileLocations(
file_paths=file_locations_list)
assert file_locations_trait.get_file_location_for_frame(

3
tests/client/ayon_core/pipeline/traits/test_time_traits.py
View file

@ -15,6 +15,7 @@ from ayon_core.pipeline.traits import (
from ayon_core.pipeline.traits.trait import TraitValidationError
def test_sequence_validations() -> None:
"""Test Sequence trait validation."""
file_locations_list = [
@ -54,7 +55,7 @@ def test_sequence_validations() -> None:
frame_spec="1001-1010,1015-1020,1100")
])
representation.get_trait(Sequence).validate(representation)
representation.get_trait(Sequence).validate_trait(representation)
# here we set handles and set them as inclusive, so this should pass
representation = Representation(name="test_2", traits=[

5
tests/client/ayon_core/pipeline/traits/test_traits.py
View file

@ -16,7 +16,7 @@ from ayon_core.pipeline.traits import (
TraitBase,
)
REPRESENTATION_DATA = {
REPRESENTATION_DATA: dict = {
FileLocation.id: {
"file_path": Path("/path/to/file"),
"file_size": 1024,
@ -127,7 +127,8 @@ def test_representation_traits(representation: Representation) -> None:
repre_dict
assert representation.has_traits() is True
empty_representation = Representation(name="test", traits=[])
empty_representation: Representation = Representation(
name="test", traits=[])
assert empty_representation.has_traits() is False
assert representation.contains_trait(trait=FileLocation) is True