FIX: Bug in bento iteration

Author: nmacholl

databento/common/bento.py

@@ -346,11 +346,10 @@ def __init__(self, data_source: DataSource) -> None:
         ] = {}
 
     def __iter__(self) -> Generator[np.void, None, None]:
+        reader = self.reader
         for _ in range(self.record_count):
-            raw = self.reader.read(self.record_size)
+            raw = reader.read(self.record_size)
             rec = np.frombuffer(raw, dtype=STRUCT_MAP[self.schema])
-            if rec.size == 0:
-                raise StopIteration
             yield rec[0]
 
     def _apply_pretty_ts(self, df: pd.DataFrame) -> pd.DataFrame:

tests/test_historical_bento.py

@@ -621,9 +621,9 @@ def test_bento_iterable() -> None:
         "22993, 1170352)"
     )
     assert (
-        str(record_list[1]) == "(14, 160, 1, 5482, 1609160400000429831, 647784973705, "
-        "3722750000000, 1, -128, 0, b'C', b'A', 1609160400000704060, "
-        "22993, 1170352)"
+        str(record_list[1]) == "(14, 160, 1, 5482, 1609160400000431665, 647784973631, "
+        "3723000000000, 1, -128, 0, b'C', b'A', 1609160400000711344, "
+        "19621, 1170353)"
     )
 
 
@@ -651,14 +651,14 @@ def test_bento_iterable_parallel() -> None:
         "22993, 1170352)"
     )
     assert (
-        str(next(second)) == "(14, 160, 1, 5482, 1609160400000429831, 647784973705, "
-        "3722750000000, 1, -128, 0, b'C', b'A', 1609160400000704060, "
-        "22993, 1170352)"
+        str(next(second)) == "(14, 160, 1, 5482, 1609160400000431665, 647784973631, "
+        "3723000000000, 1, -128, 0, b'C', b'A', 1609160400000711344, "
+        "19621, 1170353)"
     )
     assert (
-        str(next(first)) == "(14, 160, 1, 5482, 1609160400000429831, 647784973705, "
-        "3722750000000, 1, -128, 0, b'C', b'A', 1609160400000704060, "
-        "22993, 1170352)"
+        str(next(first)) == "(14, 160, 1, 5482, 1609160400000431665, 647784973631, "
+        "3723000000000, 1, -128, 0, b'C', b'A', 1609160400000711344, "
+        "19621, 1170353)"
     )