コード例 #1
0
ファイル: test_models.py プロジェクト: blady001/ImgEncrypter
    def test_get_bit_pairs(self):
        # T  ->  0b1010100
        # e  ->  0b1100101
        # s  ->  0b1110011
        # t  ->  0b1110100
        #    ->  0b100000
        # t  ->  0b1110100
        # e  ->  0b1100101
        # x  ->  0b1111000
        # t  ->  0b1110100
        text = 'Test text'

        expected_output = (
            '01', '01', '01', '00',
            '01', '10', '01', '01',
            '01', '11', '00', '11',
            '01', '11', '01', '00',
            '00', '10', '00', '00',
            '01', '11', '01', '00',
            '01', '10', '01', '01',
            '01', '11', '10', '00',
            '01', '11', '01', '00',
        )

        sample_img = np.zeros((5, 5, 3), np.uint8)
        obj = Encrypter(sample_img, text)
        pairs = obj.get_bit_pairs(text)
        self.assertEqual(pairs, expected_output)