lavc/tests/golomb: Add unit test for set_ue_golomb_long.

Add unit test for set_ue_golomb_long.

Signed-off-by: Jun Zhao <jun.zhao@intel.com>
Signed-off-by: Michael Niedermayer <michael@niedermayer.cc>
This commit is contained in:
Jun Zhao 2017-06-14 10:42:36 +08:00 committed by Michael Niedermayer
parent e61abe2d73
commit 32deea87c1
1 changed files with 19 additions and 0 deletions

View File

@ -21,6 +21,7 @@
#include <stdint.h>
#include <stdio.h>
#include "libavutil/internal.h"
#include "libavutil/mem.h"
#include "libavcodec/get_bits.h"
@ -76,6 +77,24 @@ int main(void)
}
}
#define EXTEND_L(i) ((i) << 4 | (i) & 15)
init_put_bits(&pb, temp, SIZE);
for (i = 0; i < COUNT; i++)
set_ue_golomb_long(&pb, EXTEND_L(i));
flush_put_bits(&pb);
init_get_bits(&gb, temp, 8 * SIZE);
for (i = 0; i < COUNT; i++) {
int j, s = show_bits_long(&gb, 32);
j = get_ue_golomb_long(&gb);
if (j != EXTEND_L(i)) {
fprintf(stderr, "get_ue_golomb_long: expected %d, got %d. "
"bits: %8x\n", EXTEND_L(i), j, s);
ret = 1;
}
}
init_put_bits(&pb, temp, SIZE);
for (i = 0; i < COUNT; i++)
set_se_golomb(&pb, i - COUNT / 2);