"""Annex-B access-unit split."""
from annexb import AnnexBSplitter, is_keyframe, nal_unit_type


def test_splitter_emits_complete_nal():
    s = AnnexBSplitter()
    assert s.push(b"") == []
    first = b"\x00\x00\x00\x01\x67SPS"
    second = b"\x00\x00\x00\x01\x68PPS"
    assert s.push(first + second) == [first]
    assert s.push(b"\x00\x00\x00\x01\x65IDR") == [second]


def test_is_keyframe_idr():
    au = b"\x00\x00\x00\x01\x67\x00\x00\x00\x01\x65\x88"
    assert is_keyframe(au) is True
    assert nal_unit_type(au) == 7


def test_is_keyframe_delta():
    au = b"\x00\x00\x00\x01\x41\x00"
    assert is_keyframe(au) is False
    assert nal_unit_type(au) == 1