mirror of https://bitbucket.org/ausocean/av.git
139 lines
2.9 KiB
Go
139 lines
2.9 KiB
Go
/*
|
|
DESCRIPTION
|
|
cavlc_test.go provides testing for functionality in cavlc.go.
|
|
|
|
AUTHORS
|
|
Saxon A. Nelson-Milton <saxon@ausocean.org>
|
|
|
|
LICENSE
|
|
Copyright (C) 2019 the Australian Ocean Lab (AusOcean).
|
|
|
|
It is free software: you can redistribute it and/or modify them
|
|
under the terms of the GNU General Public License as published by the
|
|
Free Software Foundation, either version 3 of the License, or (at your
|
|
option) any later version.
|
|
|
|
It is distributed in the hope that it will be useful, but WITHOUT
|
|
ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
|
FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
|
|
for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
in gpl.txt. If not, see http://www.gnu.org/licenses.
|
|
*/
|
|
|
|
package h264dec
|
|
|
|
import (
|
|
"bytes"
|
|
"reflect"
|
|
"testing"
|
|
|
|
"bitbucket.org/ausocean/av/codec/h264/h264dec/bits"
|
|
)
|
|
|
|
func TestFormCoeffTokenMap(t *testing.T) {
|
|
tests := []struct {
|
|
in [][]string
|
|
want [nColumns]map[int]map[int][2]int
|
|
}{
|
|
{
|
|
in: [][]string{
|
|
{"0", "0", "1", "11", "1111", "0000 11", "01", "1"},
|
|
{"0", "1", "0001 01", "0010 11", "0011 11", "0000 00", "0001 11", "0001 111"},
|
|
},
|
|
want: [nColumns]map[int]map[int][2]int{
|
|
0: {
|
|
0: {1: {0, 0}},
|
|
3: {5: {0, 1}},
|
|
},
|
|
1: {
|
|
0: {3: {0, 0}},
|
|
2: {11: {0, 1}},
|
|
},
|
|
2: {
|
|
0: {15: {0, 0}},
|
|
2: {15: {0, 1}},
|
|
},
|
|
3: {
|
|
4: {3: {0, 0}},
|
|
6: {0: {0, 1}},
|
|
},
|
|
4: {
|
|
1: {1: {0, 0}},
|
|
3: {7: {0, 1}},
|
|
},
|
|
5: {
|
|
0: {1: {0, 0}},
|
|
3: {15: {0, 1}},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
in: [][]string{
|
|
{"0", "0", "1", "11", "1111", "0000 11", "01", "1"},
|
|
{"0", "1", "0001 01", "0010 11", "0011 11", "-", "0001 11", "0001 111"},
|
|
},
|
|
want: [nColumns]map[int]map[int][2]int{
|
|
0: {
|
|
0: {1: {0, 0}},
|
|
3: {5: {0, 1}},
|
|
},
|
|
1: {
|
|
0: {3: {0, 0}},
|
|
2: {11: {0, 1}},
|
|
},
|
|
2: {
|
|
0: {15: {0, 0}},
|
|
2: {15: {0, 1}},
|
|
},
|
|
3: {
|
|
4: {3: {0, 0}},
|
|
},
|
|
4: {
|
|
1: {1: {0, 0}},
|
|
3: {7: {0, 1}},
|
|
},
|
|
5: {
|
|
0: {1: {0, 0}},
|
|
3: {15: {0, 1}},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
for i, test := range tests {
|
|
m, err := formCoeffTokenMap(test.in)
|
|
if err != nil {
|
|
t.Errorf("did not expect error: %v for test: %d", err, i)
|
|
}
|
|
|
|
if !reflect.DeepEqual(m, test.want) {
|
|
t.Errorf("did not get expected result for test: %d\nGot: %v\nWant: %v\n", i, m, test.want)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestParseLevelPrefix(t *testing.T) {
|
|
tests := []struct {
|
|
in string
|
|
want int
|
|
}{
|
|
{in: "00001", want: 4},
|
|
{in: "0000001", want: 6},
|
|
{in: "1", want: 0},
|
|
}
|
|
|
|
for i, test := range tests {
|
|
s, _ := binToSlice(test.in)
|
|
l, err := parseLevelPrefix(bits.NewBitReader(bytes.NewReader(s)))
|
|
if err != nil {
|
|
t.Errorf("did not expect error: %v, for test %d", err, i)
|
|
}
|
|
|
|
if l != test.want {
|
|
t.Errorf("did not get expected result for test %d\nGot: %d\nWant: %d\n", i, l, test.want)
|
|
}
|
|
}
|
|
}
|