mirror of
https://github.com/ManimCommunity/manim.git
synced 2026-06-22 10:01:47 +00:00
15 lines
486 B
Python
15 lines
486 B
Python
import numpy as np
|
|
import pytest
|
|
|
|
from manim import ImageMobject
|
|
|
|
|
|
@pytest.mark.parametrize("dtype", [np.uint8, np.uint16])
|
|
def test_invert_image(dtype):
|
|
rng = np.random.default_rng()
|
|
array = (np.iinfo(dtype).max * rng.random((10, 10, 4))).astype(dtype)
|
|
image = ImageMobject(array, pixel_array_dtype=dtype, invert=True)
|
|
assert image.pixel_array.dtype == dtype
|
|
|
|
array[:, :, :3] = np.iinfo(dtype).max - array[:, :, :3]
|
|
assert np.allclose(array, image.pixel_array)
|