from unittest import TestCase
+from hypothesis import given
+from hypothesis.strategies import binary
+
from pyac import DecodeError
from pyac import dumps
from pyac import loads
from pyac import NotEnoughData
+from pyac import Raw
class TestUnknownType(TestCase):
with self.assertRaises(NotEnoughData) as err:
loads(b"")
self.assertEqual(err.exception.n, 1)
+
+
+class TestRaw(TestCase):
+ @given(binary(min_size=1, max_size=1), binary(max_size=8))
+ def runTest(self, hdr: bytes, body: bytes) -> None:
+ self.assertSequenceEqual(dumps(Raw(hdr[0], body)), hdr + body)