def test_chi(self):
a = StateArray(initialstate)
at = chi(a)
#Checking x = 0 for every row
for y, z in itertools.product(range(5), range(a.w)):
val = (not a.bit(1,y,z)) and (a.bit(2,y,z))
self.assertEqual(at.bit(0,y,z), a.bit(0,y,z) ^ val)
def test_chi(self):
a = StateArray(initialstate)
at = chi(a)
#Checking x = 0 for every row
for y, z in itertools.product(range(5), range(a.w)):
val = (not a.bit(1, y, z)) and (a.bit(2, y, z))
self.assertEqual(at.bit(0, y, z), a.bit(0, y, z) ^ val)
def test_ro(self):
a = StateArray(initialstate)
at = ro(a)
offsets = [[0, 36, 3, 105, 210], [1, 300, 10, 45, 66],
[190, 6, 171, 15, 253], [28, 55, 153, 21, 120],
[91, 276, 231, 136, 78]]
for x in range(5):
for y in range(5):
offset = offsets[x][y]
for z in range(a.w):
self.assertEqual(a.bit(x, y, z),
at.bit(x, y, (z + offset) % at.w))
def test_ro(self):
a = StateArray(initialstate)
at = ro(a)
offsets = [[0, 36, 3, 105, 210],
[1, 300, 10, 45, 66],
[190, 6, 171, 15, 253],
[28, 55, 153, 21, 120],
[91, 276, 231, 136, 78]]
for x in range(5):
for y in range(5):
offset = offsets[x][y]
for z in range(a.w):
self.assertEqual(a.bit(x,y,z), at.bit(x,y,(z+offset)%at.w))
def test_theta(self):
a = StateArray(initialstate)
at = theta(a)
x = 3
y = 2
z = 1
c1 = a.bit(x-1,0,z) \
^ a.bit(x-1,1,z) \
^ a.bit(x-1,2,z) \
^ a.bit(x-1,3,z) \
^ a.bit(x-1,4,z)
c2 = a.bit(x+1,0,z-1) \
^ a.bit(x+1,1,z-1) \
^ a.bit(x+1,2,z-1) \
^ a.bit(x+1,3,z-1) \
^ a.bit(x+1,4,z-1)
expected = a.bit(x,y,z) ^ c1 ^ c2
self.assertEqual(at.bit(x,y,z), expected)
def test_theta(self):
a = StateArray(initialstate)
at = theta(a)
x = 3
y = 2
z = 1
c1 = a.bit(x-1,0,z) \
^ a.bit(x-1,1,z) \
^ a.bit(x-1,2,z) \
^ a.bit(x-1,3,z) \
^ a.bit(x-1,4,z)
c2 = a.bit(x+1,0,z-1) \
^ a.bit(x+1,1,z-1) \
^ a.bit(x+1,2,z-1) \
^ a.bit(x+1,3,z-1) \
^ a.bit(x+1,4,z-1)
expected = a.bit(x, y, z) ^ c1 ^ c2
self.assertEqual(at.bit(x, y, z), expected)
