From b50083809509fe69028188f8445e9c6a71ee3d6d Mon Sep 17 00:00:00 2001 From: jo2-holdsworth <jack2.holdsworth@live.uwe.ac.uk> Date: Tue, 7 Mar 2023 15:24:34 +0000 Subject: [PATCH] assert --- assert_tests.py | 8 ++++++++ main.py | 5 +++-- morseunit.py | 8 ++++++++ 3 files changed, 19 insertions(+), 2 deletions(-) create mode 100644 assert_tests.py create mode 100644 morseunit.py diff --git a/assert_tests.py b/assert_tests.py new file mode 100644 index 0000000..5e61e76 --- /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 533fa29..9d0aa48 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 0000000..bd4c3d0 --- /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 -- GitLab