t/t-bytes.py: Check that indexing, slicing, etc. return `C.ByteString'.

[catacomb-python] / t / t-algorithms.py

diff --git a/t/t-algorithms.py b/t/t-algorithms.py
index fc6bff2e83d29bbfb7b71d1bb62c1bbddf415147..d7b97cbc403a6ee6cc31f5ec3ea4d1ad2431a4e9 100644 (file)
--- a/t/t-algorithms.py
+++ b/t/t-algorithms.py
@@ -742,6 +742,13 @@ class TestKeccak (HashBufferTestMixin):
     st1.mix(m0).step()
     me.assertNotEqual(st0.extract(32), st1.extract(32))
 
     st1.mix(m0).step()
     me.assertNotEqual(st0.extract(32), st1.extract(32))
 

+    ## Check state copying.
+    st1 = st0.copy()
+    mask = st1.extract(len(m1))
+    st0.mix(m1)
+    st1.mix(m1)
+    me.assertEqual(st0.extract(32), st1.extract(32))
+

     ## Check error conditions.
     _ = st0.extract(200)
     me.assertRaises(ValueError, st0.extract, 201)
     ## Check error conditions.
     _ = st0.extract(200)
     me.assertRaises(ValueError, st0.extract, 201)


@@ -781,7 +788,7 @@ class TestKeccak (HashBufferTestMixin):
 
     ## Check masking.
     x = xcls().hash(m).xof()
 
     ## Check masking.
     x = xcls().hash(m).xof()

-    me.assertEqual(x.mask(m), C.ByteString(m) ^ C.ByteString(h[0:len(m)]))
+    me.assertEqual(x.mask(m), m ^ h[0:len(m)])

 
     ## Check the `check' method.
     me.assertTrue(xcls().hash(m).check(h0))
 
     ## Check the `check' method.
     me.assertTrue(xcls().hash(m).check(h0))