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:]))