mirror of
https://git.k-space.ee/arti/vt1200.git
synced 2026-11-02 12:23:17 +02:00
32 lines
865 B
Python
32 lines
865 B
Python
import numpy as np
|
|
from pipeline.frames import temporal_median3, box_blur3, binarize, pack_lsb
|
|
|
|
|
|
def test_temporal_median_removes_single_frame_flash():
|
|
a = np.zeros((4, 4), dtype=np.uint8)
|
|
b = np.zeros((4, 4), dtype=np.uint8)
|
|
b[1, 1] = 255
|
|
c = np.zeros((4, 4), dtype=np.uint8)
|
|
out = temporal_median3([a, b, c])
|
|
assert out[1, 1] == 0
|
|
|
|
|
|
def test_pack_lsb_is_leftmost_pixel_bit_zero():
|
|
mask = np.zeros((1, 8), dtype=bool)
|
|
mask[0, 0] = True
|
|
assert pack_lsb(mask) == b"\x01"
|
|
mask[0, 7] = True
|
|
assert pack_lsb(mask) == b"\x81"
|
|
|
|
|
|
def test_binarize_black_is_true():
|
|
gray = np.array([[0, 255]], dtype=np.uint8)
|
|
assert binarize(gray).tolist() == [[True, False]]
|
|
|
|
|
|
def test_box_blur_softens_a_single_pixel():
|
|
gray = np.zeros((3, 3), dtype=np.uint8)
|
|
gray[1, 1] = 9
|
|
out = box_blur3(gray)
|
|
assert out[1, 1] == 1
|