def test_bitset_set_unset_and_if_bitset_set(self):
b = BitSet(8)
b.bitset_set(1)
b.bitset_set(7)
assert b.bitset_as_string() == "A"
b.bitset_set(6)
assert b.bitset_as_string() == "C"
assert b.bitset_is_set(6) is True
b.bitset_unset(6)
assert b.bitset_as_string() == "A"
assert b.bitset_is_set(6) is False
assert b.bitset_is_set(2) is False
def test_bitset_valid_initialization(self):
b = BitSet(255)
assert b.size() == 255
bs = b.bitset_as_string()
# ugly mangling to get to byte_size and not recalculate
assert bs == (chr(0) * b._BitSet__byte_size)
