panda3d/tests/putil/test_bitarray.py
rdb 9d8c523dfa putil: Assorted improvements to BitArray, SparseArray, *BitMask*:
* Support converting BitMask types to int
* BitArray constructor accepts a Python long of arbitrary size
* DoubleBitMask (and QuadBitMask, by extension) supports Python long in constructor
* Support for pickling (except DoubleBitMask)
* All of them now properly define __bool__()
* More unit tests

Fixes #886
2020-04-01 20:31:16 +02:00

61 lines
1.6 KiB
Python

from panda3d.core import BitArray
import pickle
import pytest
def test_bitarray_constructor():
assert BitArray().is_zero()
assert BitArray(0).is_zero()
ba = BitArray(0x10000000000000000000000000)
assert ba.get_lowest_on_bit() == 100
assert ba.get_highest_on_bit() == 100
with pytest.raises(Exception):
assert BitArray(-1)
with pytest.raises(Exception):
assert BitArray(-10000000000000000000)
def test_bitarray_allon():
assert BitArray.all_on().is_all_on()
def test_bitarray_nonzero():
assert not BitArray()
assert not BitArray(0)
assert BitArray(1)
assert BitArray.all_on()
def test_bitarray_invert():
assert ~BitArray(0) != BitArray(0)
assert (~BitArray(0)).is_all_on()
assert ~~BitArray(0) == BitArray(0)
assert ~~BitArray(123) == BitArray(123)
def test_bitarray_getstate():
assert BitArray().__getstate__() == 0
assert BitArray(0).__getstate__() == 0
assert BitArray(100).__getstate__() == 100
assert BitArray(9870000000000000000).__getstate__() == 9870000000000000000
assert BitArray.all_on().__getstate__() == -1
assert ~BitArray(100).__getstate__() == ~100
assert ~BitArray(812000000000000000).__getstate__() == ~812000000000000000
def test_bitarray_pickle():
ba = BitArray()
assert ba == pickle.loads(pickle.dumps(ba, -1))
ba = BitArray(0)
assert ba == pickle.loads(pickle.dumps(ba, -1))
ba = BitArray(123)
assert ba == pickle.loads(pickle.dumps(ba, -1))
ba = BitArray(94187049178237918273981729127381723)
assert ba == pickle.loads(pickle.dumps(ba, -1))