diff --git a/assert_tests.py b/assert_tests.py new file mode 100644 index 0000000000000000000000000000000000000000..5e61e762d4595fd2cd8b4bcce2bc83939862fe89 --- /dev/null +++ b/assert_tests.py @@ -0,0 +1,8 @@ +import morse +def test_encode_us(): + print(morse.encode('us')) + assert morse.encode('us') == '..- ... ', "Should be ..- ..." + +if __name__ == "__main__": + test_encode_us() + print('Everything passed') \ No newline at end of file diff --git a/main.py b/main.py index 533fa29823cfd2d23c5d488e2745fa3fc0e11eb3..9d0aa489bb6a1cd45f223648b7e10b5b92254184 100644 --- a/main.py +++ b/main.py @@ -2,6 +2,7 @@ import morse if __name__ == "__main__": e = morse.encode('us') - print(e) + print('%s' % e) d = morse.decode(e) - print(d) + assert morse.encode('us') == '..- ...', "Should be ..-" + assert morse.decode('..- ...') == 'us', "Should be ..-" diff --git a/morseunit.py b/morseunit.py new file mode 100644 index 0000000000000000000000000000000000000000..bd4c3d07af5a3ca07eeda45773fcb7ab3e090c1d --- /dev/null +++ b/morseunit.py @@ -0,0 +1,8 @@ +import unittest +import morse +class TestMorse(unittest.TestCase): + def test_encode_us(self): + self.assertEqual( morse.encode('us'), '..- ... ') + +if __name__ == '__main__': + unittest.main() \ No newline at end of file