# -*- coding: utf-8 -*-

from constant import HW_ACCEL_CUDA, HW_ACCEL_NONE, HW_ACCEL_QSV
from handlers.base import get_video_encode_args


def _assert_bframe_disabled(args):
    assert '-bf' in args
    bf_index = args.index('-bf')
    assert args[bf_index + 1] == '0'


def test_get_video_encode_args_disable_b_frames_for_software():
    args = get_video_encode_args(HW_ACCEL_NONE)
    _assert_bframe_disabled(args)


def test_get_video_encode_args_disable_b_frames_for_qsv():
    args = get_video_encode_args(HW_ACCEL_QSV)
    _assert_bframe_disabled(args)


def test_get_video_encode_args_disable_b_frames_for_cuda():
    args = get_video_encode_args(HW_ACCEL_CUDA)
    _assert_bframe_disabled(args)