You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
59 lines
1.2 KiB
59 lines
1.2 KiB
from lib import util
|
|
|
|
|
|
def test_cachedproperty():
|
|
class Target:
|
|
|
|
CALL_COUNT = 0
|
|
|
|
def __init__(self):
|
|
self.call_count = 0
|
|
|
|
@util.cachedproperty
|
|
def prop(self):
|
|
self.call_count += 1
|
|
return self.call_count
|
|
|
|
@util.cachedproperty
|
|
def cls_prop(cls):
|
|
cls.CALL_COUNT += 1
|
|
return cls.CALL_COUNT
|
|
|
|
|
|
t = Target()
|
|
assert t.prop == t.prop == 1
|
|
assert Target.cls_prop == Target.cls_prop == 1
|
|
|
|
|
|
def test_deep_getsizeof():
|
|
int_t = util.deep_getsizeof(1)
|
|
assert util.deep_getsizeof([1, 1]) > 2 * int_t
|
|
assert util.deep_getsizeof({1: 1}) > 2 * int_t
|
|
assert util.deep_getsizeof({1: {1: 1}}) > 3 * int_t
|
|
|
|
|
|
class Base:
|
|
pass
|
|
|
|
|
|
class A(Base):
|
|
pass
|
|
|
|
|
|
class B(Base):
|
|
pass
|
|
|
|
|
|
def test_subclasses():
|
|
assert util.subclasses(Base) == [A, B]
|
|
assert util.subclasses(Base, strict=False) == [A, B, Base]
|
|
|
|
|
|
def test_chunks():
|
|
assert list(util.chunks([1, 2, 3, 4, 5], 2)) == [[1, 2], [3, 4], [5]]
|
|
|
|
|
|
def test_increment_byte_string():
|
|
assert util.increment_byte_string(b'1') == b'2'
|
|
assert util.increment_byte_string(b'\x01\x01') == b'\x01\x02'
|
|
assert util.increment_byte_string(b'\xff\xff') == b'\x01\x00\x00'
|
|
|