Sunpy/sampy3
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Make use of yield in __iter__

Browse Source
This commit is contained in:
Emily 2023-03-15 03:33:24 +01:00
parent 73eaa0e89b
commit 404f71dddd
2 changed files with 3 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
sampy/raknet/bitstream.py
View File

@ -167,15 +167,8 @@ class Bitstream:
return self
def __iter__(self) -> Bitstream:
self._iter_index = 0
return self
def __next__(self) -> bool:
if self._iter_index >= self._bit_length:
raise StopIteration
bit = self[self._iter_index]
self._iter_index += 1
return bit
for i in range(len(self)):
yield self[i]
def __repr__(self) -> str:
return "<Bitstream addr:0x%012x offset:%d len:%d data:'%s'>" % (

2
tests/test_bitstream.py
View File

@ -181,7 +181,7 @@ def test_add(
])
def test_iter(init_values: List[Union[bool, List[bool], bytes, str, Bitstream]]):
bitstream = Bitstream(*init_values)
assert len(iter(bitstream)) == len(bitstream)
assert len(list(bitstream)) == len(bitstream)
for index, bit in enumerate(bitstream):
assert bit == bitstream[index]