chapter10

#!/usr/bin/env python
# -*- coding: utf-8 -*-
class TSzyfrant:

    def __init__(self,klucz):
        self.klucz=klucz

    def szyfruj(self, liczba):
        return (liczba + self.klucz) % 256

    def odszyfruj(self, liczba):
        return (256 + liczba - self.klucz) % 256

def TestFunkcjonalny():
    szyfrant=TSzyfrant(88)
    wiadomosc = int(input('Podaj liczbę: '))
    szyfr=szyfrant.szyfruj(wiadomosc)
    print('Kod=%s' % szyfr)
    print('Odszyfrowano=%s' % szyfrant.odszyfruj(szyfr))

import unittest
from random import randrange

class TestyJednostkowe(unittest.TestCase):

    def setUp(self):
        """Call before every test case."""
        self.szyfrant = TSzyfrant(88)

    def test1(self):
        for n in range(10):
            msg = randrange(255)
            szyfr = self.szyfrant.szyfruj(msg)
            print(szyfr)
            msg2 = self.szyfrant.odszyfruj(szyfr)
            self.assertEqual(msg, msg2)

import sys

if __name__ == '__main__':
  runner = unittest.TextTestRunner(stream=sys.stdout, verbosity=2)
  unittest.main(testRunner=runner, exit=False)
  # TestFunkcjonalny()
Kodowanie oparte o testy
