def test_get_node_count_big(self):
bs = GhettoBitStream([])
# count bigger than 127, need next 2 bytes to represent it
bs.bytes = deque([130, 1, 17])
ret = PathTree._get_node_count(bs)
self.assertEqual(ret, 273)
def test_get_node_count_medium(self):
bs = GhettoBitStream([])
# count bigger than 127, only need 1 byte to represent it
bs.bytes = deque([129, 150])
ret = PathTree._get_node_count(bs)
self.assertEqual(ret, 150)
def test_get_node_count_small(self):
bs = GhettoBitStream([])
bs.bytes = deque([6])
ret = PathTree._get_node_count(bs)
self.assertEqual(ret, 6)
def test_get_node_count_big(self):
bs = GhettoBitStream([])
# count bigger than 127, need next 2 bytes to represent it
bs.bytes = deque([130, 1, 17])
ret = PathTree._get_node_count(bs)
self.assertEqual(ret, 273)
def test_get_node_count_medium(self):
bs = GhettoBitStream([])
# count bigger than 127, only need 1 byte to represent it
bs.bytes = deque([129, 150])
ret = PathTree._get_node_count(bs)
self.assertEqual(ret, 150)
def test_get_node_count_small(self):
bs = GhettoBitStream([])
bs.bytes = deque([6])
ret = PathTree._get_node_count(bs)
self.assertEqual(ret, 6)
