def test_convert_to_bytes_string(self):
char = 'a' # 7 bits
py_bin = bin(ord(char))
self.assertEqual(len(py_bin[2:]), 7)
my_bin = Encrypter.convert_to_bytes_string(char)
self.assertEqual(len(my_bin), 8)
self.assertEqual(py_bin[2:], my_bin[1:])
def test_end_message_found(self):
sample_img = np.zeros((5, 5, 3), np.uint8)
decrypter = Decrypter(sample_img)
msg = '0110001001010011' # just random
self.assertFalse(decrypter.end_message_found(msg))
msg = 'test' + Decrypter.ENDMSG_CODE
bin_msg = ''
for x in msg:
binary = Encrypter.convert_to_bytes_string(x)
bin_msg += binary
self.assertTrue(decrypter.end_message_found(bin_msg[-64:]))
