Merge pull request #12 from mortbauer/master

do not use assert since otherwise running with -O will fail
d9pouces · Jan 11, 2025 · 700afd1 · 700afd1
2 parents 0eef2a4 + 55f64b0
commit 700afd1
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/rubymarshal/reader.py b/rubymarshal/reader.py
@@ -282,8 +282,10 @@ def read_symreal(self):
 
 
 def load(fd, registry=None):
-    assert fd.read(1) == b"\x04"
-    assert fd.read(1) == b"\x08"
+    if fd.read(1) != b"\x04":
+        raise ValueError(r'Expected token \x04')
+    if fd.read(1) != b"\x08":
+        raise ValueError(r'Expected token \x08')
 
     loader = Reader(fd, registry=registry)
     return loader.read()